small_fix

shiyutang · Jan 11, 2024 · ac8c37a · ac8c37a
1 parent e9655d3
commit ac8c37a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/erniebot-agent/tests/integration_tests/agents/test_agent_with_plugins.py b/erniebot-agent/tests/integration_tests/agents/test_agent_with_plugins.py
@@ -119,7 +119,7 @@ async def __call__(self, location, unit: int = None) -> None:
 
 
 # TODO(shiyutang): replace this when model is online
-llm = ERNIEBot(model="ernie-3.5", api_type="custom", enable_multi_step_tool_call=True)
+llm = ERNIEBot(model="ernie-3.5", api_type="qianfan", enable_multi_step_tool_call=True)
 memory = SlidingWindowMemory(max_round=1)
 plugins = ["ChatFile", "eChart"]
 # plugins: List[str] = []
@@ -135,11 +135,11 @@ async def __call__(self, location, unit: int = None) -> None:
 
 
 async def run_agent():
-    await GlobalFileManagerHandler().configure(
+    GlobalFileManagerHandler().configure(
         enable_remote_file=True,
         access_token="your-access-token",
     )
-    file_manager = await GlobalFileManagerHandler().get()
+    file_manager = GlobalFileManagerHandler().get()
 
     docx_file = await file_manager.create_file_from_path(
         file_path="浅谈牛奶的营养与消费趋势.docx",

diff --git a/erniebot/src/erniebot/resources/chat_completion_with_plugins.py b/erniebot/src/erniebot/resources/chat_completion_with_plugins.py
@@ -46,7 +46,7 @@ class ChatCompletionWithPlugins(EBResource, CreatableWithStreaming):
     )
     _API_INFO_DICT: ClassVar[Dict[APIType, Dict[str, Any]]] = {
         APIType.QIANFAN: {
-            "path": "/erniebot/plugins",
+            "path": "/erniebot/plugin",
         },
         APIType.CUSTOM: {
             "path": "/erniebot/plugins_v3",