1) Do some rename refactor ,prepare for LLMProcess refactor

2) Fix merge bugs.
This commit is contained in:
Liu Zhicong
2023-12-06 13:31:05 -08:00
parent 35d204ac05
commit 8739bf6a76
44 changed files with 693 additions and 2043 deletions
+1 -1
View File
@@ -1,6 +1,6 @@
import os
from typing import Any,List,Dict
from aios import AgentMsg,AgentTodo,AgentPrompt
from aios import AgentMsg,AgentTodo,LLMPrompt
from aios import SimpleAIFunction, SimpleAIOperation
from aios import SimpleEnvironment
+2 -2
View File
@@ -218,7 +218,7 @@ class IssueAgent(CustomAIAgent):
super().__init__(agent_id, llm_model_name, max_token_size)
class IssueParserEnvironment(Environment):
class IssueParserEnvironment(SimpleEnvironment):
def __init__(self, env_id: str, storage: IssueStorage) -> None:
super().__init__(env_id)
self.storage = storage
@@ -305,7 +305,7 @@ class IssueParser:
mail_desc = Mail.prompt_desc()
issue_desc = Issue.prompt_desc()
prompt = AgentPrompt()
prompt = LLMPrompt()
prompt.system_message = {"role": "system", "content": f'''
I'm a CEO of a company named 巴克云; You'ar my assistant, and you should help me to manage my issues. Issues is a concept in software development of this company, but I use it to manage my work.
I'll give you a mail in json format, {mail_desc};
+4 -1
View File
@@ -1,3 +1,4 @@
import asyncio
import openai
from openai import AsyncOpenAI
import os
@@ -237,7 +238,8 @@ class OpenAI_ComputeNode(ComputeNode):
return result
logger.info(f"openai response: {resp}")
if mode_name == "gpt-4-vision-preview":
#TODO: gpt-4v api is image_2_text ?
if mode_name == "gpt-4-vision-preview":
status_code = resp.choices[0].finish_reason
if status_code is None:
status_code = resp.choices[0].finish_details['type']
@@ -265,6 +267,7 @@ class OpenAI_ComputeNode(ComputeNode):
if token_usage:
result.result_refers["token_usage"] = token_usage
logger.info(f"openai success response: {result.result_str}")
return result
case _:
+6 -2
View File
@@ -87,7 +87,7 @@ class TelegramTunnel(AgentTunnel):
async def _run_app():
try:
update_id = (await self.bot.get_updates())[0].update_id
except IndexError:
except Exception as e:
update_id = None
except Exception as e:
logger.error(f"tg_tunnel error:{e}")
@@ -97,7 +97,11 @@ class TelegramTunnel(AgentTunnel):
#logger.info("listening for new messages...")
while True:
try:
update_id = await self._do_process_raw_message(self.bot, update_id)
if update_id:
update_id = await self._do_process_raw_message(self.bot, update_id)
else:
update_id = (await self.bot.get_updates())[0].update_id
except NetworkError:
await asyncio.sleep(1)
except Forbidden:
@@ -2,7 +2,7 @@ import logging
import toml
import os
from aios import AIStorage,PackageEnv,PackageEnvManager,PackageMediaInfo,PackageInstallTask
from aios import AIStorage,PackageEnv,PackageEnvManager,PackageMediaInfo,PackageInstallTask,Workflow
from agent_manager import AgentManager
logger = logging.getLogger(__name__)