gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

1) Add e-mail tunnel support 2) Modify all __new__ to get_instance() 3) init docker build scripts

Browse Source
This commit is contained in:
Liu Zhicong
2023-09-16 11:41:59 -07:00
parent ef9ab83e63
commit 43c70489e7
15 changed files with 272 additions and 70 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/workflow.py
+7 -3
View File
@@ -122,7 +122,11 @@ class Workflow:
connected_env_ndoe = config.get("connected_env")
if connected_env_ndoe is not None:
for _node in connected_env_ndoe:
remote_env = Environment.get_env_by_id(_node.get("env_id"))
env_id = _node.get("env_id")
if env_id is None:
continue
remote_env = Environment.get_env_by_id(_node.get(env_id))
if remote_env is None:
logger.error(f"Workflow load connected_env failed, env {env_id} not found!")
return False
@@ -391,7 +395,7 @@ class Workflow:
inner_functions = self._get_inner_functions()
prompt.messages.append({"role":"function","content":result_str,"name":func_name})
task_result:ComputeTaskResult = await ComputeKernel().do_llm_completion(prompt,
task_result:ComputeTaskResult = await ComputeKernel.get_instance().do_llm_completion(prompt,
the_role.agent.llm_model_name,the_role.agent.max_token_size,
inner_functions)
@@ -431,7 +435,7 @@ class Workflow:
async def _do_process_msg():
#TODO: send msg to agent might be better?
task_result:ComputeTaskResult = await ComputeKernel().do_llm_completion(prompt,the_role.agent.get_llm_model_name(),the_role.agent.get_max_token_size(),inner_functions)
task_result:ComputeTaskResult = await ComputeKernel.get_instance().do_llm_completion(prompt,the_role.agent.get_llm_model_name(),the_role.agent.get_max_token_size(),inner_functions)
result_str = task_result.result_str
logger.info(f"{the_role.role_id} process {msg.sender}:{msg.body},llm str is :{result_str}")