fix: remove the data dir parameter

dbpunk-labs · Sep 12, 2023 · 688b26c · 688b26c
1 parent c03deed
commit 688b26c
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/agent/setup.py b/agent/setup.py
@@ -38,7 +38,7 @@
         "grpc-google-iam-v1>=0.12.6",
         "aiofiles",
         "python-dotenv",
-        "openai"
+        "openai",
     ],
     entry_points={
         "console_scripts": [

diff --git a/agent/src/octopus_agent/gpt_tools.py b/agent/src/octopus_agent/gpt_tools.py
@@ -23,6 +23,8 @@
     AsyncCallbackManagerForToolRun,
     CallbackManagerForToolRun,
 )
+
+
 class AskQuestionInput(BaseModel):
     question: str = Field(description="the question")
 
@@ -56,6 +58,7 @@ class ExecutePythonCodeInput(BaseModel):
         description="the saved filename list", default=[]
     )
 
+
 class ExecutePythonCodeTool(StructuredTool):
     name = "execute_python_code"
     description = """Execute arbitrary Python code Returns a Markdown format string including return code, result, stdout, stderr, error"""

diff --git a/agent/tests/agent_tool_tests.py b/agent/tests/agent_tool_tests.py
@@ -40,7 +40,7 @@ def kernel_sdk():
 async def test_async_smoke_test(kernel_sdk):
     code = "print('hello world!')"
     sdk = kernel_sdk
-    api = OctopusAPIJsonOutput(sdk, api_data_dir)
+    api = OctopusAPIJsonOutput(sdk)
     result = await api.arun(code)
     assert not result["result"]
     assert result["stdout"] == "hello world!\n"