feat(Agent): change plugin path

change plugin path
eosphoros-ai · Dec 26, 2023 · 7192cdf · 7192cdf
1 parent 954834e
commit 7192cdf
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/dbgpt/agent/agents/agent.py b/dbgpt/agent/agents/agent.py
@@ -172,7 +172,7 @@ class AgentContext:
     resource_db: Optional[AgentResource] = None
     resource_knowledge: Optional[AgentResource] = None
     resource_internet: Optional[AgentResource] = None
-    llm_models: Optional[List[ModelMetadata]] = None
+    llm_models: Optional[List[Union[ModelMetadata, str]]] = None
     model_priority: Optional[dict] = None
     agents: Optional[List[str]] = None
 

diff --git a/examples/agents/auto_plan_agent_dialogue_example.py b/examples/agents/auto_plan_agent_dialogue_example.py
@@ -25,6 +25,7 @@
 
 from dbgpt.agent.agents.agent import AgentContext
 from dbgpt.agent.memory.gpts_memory import GptsMemory
+from dbgpt.core.interface.llm import ModelMetadata
 
 import asyncio
 import os
@@ -34,7 +35,7 @@
 
     llm_client = OpenAILLMClient()
     context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)
-    context.llm_models = ["gpt-3.5-turbo"]
+    context.llm_models =  [ModelMetadata(model="gpt-3.5-turbo")]
 
     default_memory = GptsMemory()
     coder = CodeAssistantAgent(memory=default_memory, agent_context=context)

diff --git a/examples/agents/single_agent_dialogue_example.py b/examples/agents/single_agent_dialogue_example.py
@@ -18,6 +18,7 @@
 from dbgpt.agent.agents.user_proxy_agent import UserProxyAgent
 from dbgpt.agent.memory.gpts_memory import GptsMemory
 from dbgpt.agent.agents.agent import AgentContext
+from dbgpt.core.interface.llm import ModelMetadata
 import asyncio
 import os
 
@@ -28,7 +29,7 @@
 
     llm_client = OpenAILLMClient()
     context: AgentContext = AgentContext(conv_id="test456", llm_provider=llm_client)
-    context.llm_models = ["gpt-3.5-turbo"]
+    context.llm_models = [ModelMetadata(model="gpt-3.5-turbo")]
 
     default_memory = GptsMemory()
     coder = CodeAssistantAgent(memory=default_memory, agent_context=context)