skills/copilotkit__copilotkit
1
0
Fork 0
mirror of https://github.com/CopilotKit/CopilotKit.git synced 2026-09-14 16:26:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

use AsyncTavilyClient everywhere in python code

Browse Source
This commit is contained in:
Markus Ecker
2024-12-03 13:59:29 +01:00
parent a5efccd69d
commit dd48edfe0f
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
sdk-python/copilotkit/demos/research_canvas/search.py
+3 -3
View File
@@ -8,7 +8,7 @@ from pydantic import BaseModel, Field
from langchain_core.runnables import RunnableConfig
from langchain_core.messages import AIMessage, ToolMessage, SystemMessage
from langchain.tools import tool
from tavily import TavilyClient
from tavily import AsyncTavilyClient
from copilotkit.langchain import copilotkit_emit_state, copilotkit_customize_config
from copilotkit.demos.research_canvas.state import AgentState
from copilotkit.demos.research_canvas.model import get_model
@@ -23,7 +23,7 @@ class ResourceInput(BaseModel):
def ExtractResources(resources: List[ResourceInput]): # pylint: disable=invalid-name,unused-argument
"""Extract the 3-5 most relevant resources from a search result."""
tavily_client = TavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
tavily_client = AsyncTavilyClient(api_key=os.getenv("TAVILY_API_KEY"))
async def search_node(state: AgentState, config: RunnableConfig):
"""
@@ -46,7 +46,7 @@ async def search_node(state: AgentState, config: RunnableConfig):
search_results = []
for i, query in enumerate(queries):
response = tavily_client.search(query)
response = await tavily_client.search(query)
search_results.append(response)
state["logs"][i]["done"] = True
await copilotkit_emit_state(config, state)