Fix examples and Build Full Pipeline

Author: CongkeChen

examples/langgraph_agent/agent/agent.py

@@ -39,8 +39,9 @@ def build_calculator_subgraph():
     api_key, url, model_name = get_model_config()
     model = init_chat_model(
         model_name,
+        model_provider="openai",
         api_key=api_key,
-        api_base=url,
+        base_url=url,
     )
 
     tools = [calculate]
@@ -79,8 +80,9 @@ def build_graph_with_subgraph():
     api_key, url, model_name = get_model_config()
     model = init_chat_model(
         model_name,
+        model_provider="openai",
         api_key=api_key,
-        api_base=url,
+        base_url=url,
     )
 
     # Build calculator subgraph
@@ -126,8 +128,9 @@ def build_graph():
     api_key, url, model_name = get_model_config()
     model = init_chat_model(
         model_name,
+        model_provider="openai",
         api_key=api_key,
-        api_base=url,
+        base_url=url,
     )
 
     tools = [calculate]

examples/langgraph_agent_with_cancel/agent/agent.py

@@ -44,8 +44,9 @@ def build_graph():
     api_key, url, model_name = get_model_config()
     model = init_chat_model(
         model_name,
+        model_provider="openai",
         api_key=api_key,
-        api_base=url,
+        base_url=url,
     )
 
     tools = [calculate, analyze_data]

examples/langgraphagent_with_human_in_the_loop/agent/agent.py

@@ -53,8 +53,9 @@ def _build_graph():
     api_key, url, model_name = get_model_config()
     model = init_chat_model(
         model_name,
+        model_provider="openai",
         api_key=api_key,
-        api_base=url,
+        base_url=url,
     )
     tools = [execute_database_operation]
     llm_with_tools = model.bind_tools(tools)

examples/mem0_tools/agent/agent.py

@@ -10,8 +10,8 @@
 from trpc_agent_sdk.agents import LlmAgent
 from trpc_agent_sdk.models import LLMModel
 from trpc_agent_sdk.models import OpenAIModel
-from trpc_agent_sdk.tools.mem0_tools import SaveMemoryTool
-from trpc_agent_sdk.tools.mem0_tools import SearchMemoryTool
+from trpc_agent_sdk.tools.mem0_tool import SaveMemoryTool
+from trpc_agent_sdk.tools.mem0_tool import SearchMemoryTool
 
 from .config import get_mem0_platform_config
 from .config import get_memory_config