1) FixBug

2) Add Documents abount TASK/TODO Refactor
This commit is contained in:
Liu Zhicong
2024-01-06 13:08:41 -08:00
parent 3663f140fc
commit c975ba2053
22 changed files with 398 additions and 109 deletions
+3 -3
View File
@@ -154,7 +154,7 @@ class AgentMsg:
@staticmethod
def create_image_body(images: [str], prompt: str = None):
return json.dumps({"images": images, "prompt": prompt})
return json.dumps({"images": images, "prompt": prompt}, ensure_ascii=False)
@staticmethod
def parse_image_body(image_body: str) -> Tuple[str, List[str]]:
@@ -163,7 +163,7 @@ class AgentMsg:
@staticmethod
def create_video_body(video: str, prompt: str = None):
return json.dumps({"video": video, "prompt": prompt})
return json.dumps({"video": video, "prompt": prompt}, ensure_ascii=False)
@staticmethod
def parse_video_body(video_body: str) -> Tuple[str, str]:
@@ -172,7 +172,7 @@ class AgentMsg:
@staticmethod
def create_audio_body(audio: str, prompt: str = None):
return json.dumps({"audio": audio, "prompt": prompt})
return json.dumps({"audio": audio, "prompt": prompt}, ensure_ascii=False)
@staticmethod
def parse_audio_body(audio_body: str) -> Tuple[str, str]:
+12 -11
View File
@@ -60,7 +60,7 @@ class AIFunction:
else:
parameters_str += f"{k} (Optional): {v.description},"
if len(parameters_str) > 0:
return f"{self.get_description} Parameters: {parameters_str}"
return f"{self.get_description()} Parameters: {parameters_str}"
return f"f{self.get_description()}, no parameters"
@abstractmethod
@@ -246,12 +246,16 @@ class SimpleAIFunction(AIFunction):
class AIAction:
@abstractmethod
def get_name(self) -> str:
def get_id(self) -> str:
"""
return the name of the operation (should be snake case)
"""
pass
def get_name(self)->str:
return self.get_id().split('.')[-1].strip()
@abstractmethod
def get_description(self) -> str:
"""
@@ -274,7 +278,7 @@ class SimpleAIAction(AIAction):
self.description = description
self.func_handler = func_handler
def get_name(self) -> str:
def get_id(self) -> str:
return self.op
def get_description(self) -> str:
@@ -289,17 +293,14 @@ class SimpleAIAction(AIAction):
class AIFunction2Action(AIAction):
def __init__(self, func: AIFunction) -> None:
self.func = func
super().__init__()
self.ai_func = func
@abstractmethod
def get_name(self) -> str:
return self.func.get_id()
def get_id(self) -> str:
return self.ai_func.get_id()
@abstractmethod
def get_description(self) -> str:
return self.func.get_detail_description()
return self.ai_func.get_detail_description()
@abstractmethod
async def execute(self, params: dict) -> str:
self.func.execute(**params)
return await self.ai_func.execute(params)
+3 -3
View File
@@ -95,11 +95,11 @@ class LLMPrompt:
def as_str(self)->str:
result_str = ""
if self.system_message:
result_str = json.dumps(self.system_message)
result_str = json.dumps(self.system_message,ensure_ascii=False)
if self.messages:
result_str += json.dumps(self.messages)
result_str += json.dumps(self.messages,ensure_ascii=False)
if self.inner_functions:
result_str += json.dumps(self.inner_functions)
result_str += json.dumps(self.inner_functions,ensure_ascii=False)
return result_str