1) FixBug
2) Add Documents abount TASK/TODO Refactor
This commit is contained in:
@@ -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]:
|
||||
|
||||
@@ -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)
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user