Fix workflow system prompt bugs.

This commit is contained in:
Liu Zhicong
2023-09-19 18:25:25 -07:00
parent 298ee73a6b
commit 40e14ee025
7 changed files with 132 additions and 70 deletions
+4 -14
View File
@@ -2,6 +2,7 @@ from enum import Enum
import uuid
import time
import re
import shlex
class AgentMsgType(Enum):
TYPE_MSG = 0
@@ -123,19 +124,8 @@ class AgentMsg:
@classmethod
def parse_function_call(cls,func_string:str):
match = re.search(r'\s*(\w+)\s*\(\s*(.*)\s*\)\s*', func_string)
if not match:
return None
func_name = match.group(1)
if func_name is None:
return None
if len(func_name) < 2:
return None
params_string = match.group(2).strip()
params = re.split(r'\s*,\s*(?=(?:[^"]*"[^"]*")*[^"]*$)', params_string)
params = [param.strip('"') for param in params]
str_list = shlex.split(func_string)
func_name = str_list[0]
params = str_list[1:]
return func_name, params