fix: replace chat template with alternative roles (#77)

agiresearch · Jun 28, 2024 · 6d54bdb · 6d54bdb
1 parent 1e94868
commit 6d54bdb
Show file tree

Hide file tree

Showing 7 changed files with 14 additions and 17 deletions.
diff --git a/.github/workflows/publish.yml b/.github/workflows/publish.yml
@@ -26,8 +26,5 @@ jobs:
           pip install hatch
       - name: Build package
         run: hatch build
-      - name: Clear GZ
-        run: |
-          find dist -name "pyopenagi-*.tar.gz" -delete
       - name: Publish package distributions to PyPI
         uses: pypa/gh-action-pypi-publish@release/v1
diff --git a/pyopenagi/agents/agent_config/CreationAgent.json b/pyopenagi/agents/agent_config/CreationAgent.json
@@ -12,7 +12,7 @@
             "type": "function",
             "function": {
                 "name": "sdxl-turbo",
-                "description": "generate images by calling SDXL-turbo model",
+                "description": "generate images by calling sdxl-turbo model",
                 "parameters": {
                     "type": "object",
                     "properties": {

diff --git a/pyopenagi/agents/native_agents/academic_agent/academic_agent.py b/pyopenagi/agents/native_agents/academic_agent/academic_agent.py
@@ -83,8 +83,8 @@ def run(self):
                         function_responses += function_response
 
                         self.messages.append({
-                            "role": "user",
-                            "content": f"It calls the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
+                            "role": "assistant",
+                            "content": f"I will call the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
                         })
 
                         self.logger.log(f"At current step, it will call the {function_name} with the params as {function_args}. The tool response is {function_responses}\n", level="info")
@@ -99,7 +99,7 @@ def run(self):
 
             else:
                 self.messages.append({
-                    "role": "user",
+                    "role": "assistant",
                     "content": response_message
                 })
                 if i == len(self.workflow) - 1:

diff --git a/pyopenagi/agents/native_agents/creation_agent/creation_agent.py b/pyopenagi/agents/native_agents/creation_agent/creation_agent.py
@@ -97,8 +97,8 @@ def run(self):
 
 
                         self.messages.append({
-                            "role": "user",
-                            "content": f"It calls the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
+                            "role": "assistant",
+                            "content": f"I will call the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
                         })
 
                     except Exception:
@@ -111,7 +111,7 @@ def run(self):
 
             else:
                 self.messages.append({
-                    "role": "user",
+                    "role": "assistant",
                     "content": response_message
                 })
                 if i == len(self.workflow) - 1:

diff --git a/pyopenagi/agents/native_agents/math_agent/math_agent.py b/pyopenagi/agents/native_agents/math_agent/math_agent.py
@@ -94,8 +94,8 @@ def run(self):
                         function_responses += function_response
 
                         self.messages.append({
-                            "role": "user",
-                            "content": f"It calls the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
+                            "role": "assistant",
+                            "content": f"I will call the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
                         })
 
                         self.logger.log(f"For current step, it will call the {function_name} with the params as {function_args}. The tool response is {function_responses}\n", level="info")
@@ -110,7 +110,7 @@ def run(self):
 
             else:
                 self.messages.append({
-                    "role": "user",
+                    "role": "assistant",
                     "content": response_message
                 })
 

diff --git a/pyopenagi/agents/native_agents/rec_agent/rec_agent.py b/pyopenagi/agents/native_agents/rec_agent/rec_agent.py
@@ -83,8 +83,8 @@ def run(self):
                         function_responses += function_response
 
                         self.messages.append({
-                            "role": "user",
-                            "content": f"It calls the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
+                            "role": "assistant",
+                            "content": f"I will call the {function_name} with the params as {function_args} to solve this. The tool response is {function_responses}\n"
                         })
 
                         self.logger.log(f"For current step, it will call the {function_name} with the params as {function_args}. The tool response is {function_responses}\n", level="info")
@@ -99,7 +99,7 @@ def run(self):
 
             else:
                 self.messages.append({
-                    "role": "user",
+                    "role": "assistant",
                     "content": response_message
                 })
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -13,7 +13,7 @@ license = {file = "LICENSE"}
 name = "pyopenagi"
 readme = "README.md"
 requires-python = ">=3.9"
-version = "0.0.5"
+version = "0.0.5.beta1"
 
 classifiers = [
   "Development Status :: 3 - Alpha",