admin/cyclop
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Adjust SYSTEM_PROMPT_PATH usage to support new file selection

Browse Source
This commit is contained in:
bucolucas
2024-08-21 08:09:47 -05:00
parent b8f716d532
commit ea89f9e4fd
1 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
base_telegram_inference_bot.py
+7 -3
View File
@@ -14,8 +14,12 @@ class BaseTelegramInferenceBot(ABC):
@staticmethod @staticmethod
def load_system_prompt(): def load_system_prompt():
with open(os.environ.get("SYSTEM_PROMPT_PATH"), "r", encoding="utf-8") as file: system_prompt_path = os.getenv("SYSTEM_PROMPT_PATH")
return file.read().strip() if system_prompt_path and os.path.isfile(system_prompt_path):
with open(system_prompt_path, "r", encoding="utf-8") as file:
return file.read().strip()
else:
raise FileNotFoundError("SYSTEM_PROMPT_PATH is not set or file does not exist.")
@staticmethod @staticmethod
def load_functions(): def load_functions():
@@ -52,7 +56,7 @@ class BaseTelegramInferenceBot(ABC):
def call_tool(self, function_call_name, function_call_arguments): def call_tool(self, function_call_name, function_call_arguments):
function_name = function_call_name function_name = function_call_name
function_args = json.loads(function_call_arguments if function_call_arguments is not None else "{}") function_args = json.loads(function_call_arguments if function_call_arguments is not None else "{}").
for tool in self.tools: for tool in self.tools:
if function_name in [f["name"] for f in tool.get_functions()]: if function_name in [f["name"] for f in tool.get_functions()]:
return tool.execute(function_name, **function_args) return tool.execute(function_name, **function_args)