Доработка: 1) В agent.py реализовать функцию main() с циклом чтения пол

agent.py

@@ -80,3 +80,55 @@ def main():
 
 if __name__ == "__main__":
     main()
+```
+
+```python
+# main.py
+
+import os
+from dotenv import load_dotenv
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain.tools import tool
+from vectorstore import get_vectorstore
+
+load_dotenv()
+TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
+if not TAVILY_API_KEY:
+    raise ValueError("TAVILY_API_KEY not found in .env file")
+
+vectorstore = get_vectorstore()
+
+@tool
+def search_local_kb(query: str) -> str:
+    """Search local knowledge base using ChromaDB."""
+    retriever = vectorstore.as_retriever(search_kwargs={"k": 3})
+    docs = retriever.get_relevant_documents(query)
+    return "\n\n".join([doc.page_content for doc in docs])
+
+@tool
+def web_search(query: str) -> str:
+    """Search the web using Tavily."""
+    search = TavilySearchResults(tavily_api_key=TAVILY_API_KEY, max_results=3)
+    results = search.run(query)
+    return "\n\n".join([result["content"] for result in results])
+```
+
+```python
+# tools.py
+
+import os
+from dotenv import load_dotenv
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain.tools import tool
+
+load_dotenv()
+TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
+if not TAVILY_API_KEY:
+    raise ValueError("TAVILY_API_KEY not found in .env file")
+
+@tool
+def web_search(query: str) -> str:
+    """Search the web using Tavily."""
+    search = TavilySearchResults(tavily_api_key=TAVILY_API_KEY, max_results=3)
+    results = search.run(query)
+    return "\n\n".join([result["content"] for result in results])