8. Самописный поисковый агент на основе deep agents from scratch: agent.py

solutions/69de7223f309a98be0007e09_8__Самописный_поисковый_агент_на_основе_/agent.py

@@ -5,6 +5,8 @@ from langchain_ollama import OllamaEmbeddings, OllamaLLM
 from langchain_qdrant import QdrantVectorStore
 from langchain.schema import Document
 from langchain.tools import tool
+from rich.console import Console
+from rich.table import Table
 
 # Конфигурация подключения к Qdrant
 QDRANT_URL = os.getenv("QDRANT_URL", "http://localhost:6333")
@@ -45,9 +47,22 @@ def add_to_knowledge_base(doc: str) -> str:
 
 # Если нужно протестировать работу инструментов
 if __name__ == "__main__":
+    console = Console()
+
     # Пример добавления документа
-    print(add_to_knowledge_base("Python – это язык программирования, созданный Гвидо ван Россумом."))
+    add_msg = add_to_knowledge_base(
+        "Python – это язык программирования, созданный Гвидо ван Россумом."
+    )
+    console.print(f"[green]{add_msg}[/green]")
+
     # Пример поиска
     results = search_knowledge_base("что такое Python?")
+    table = Table(title="Результаты поиска")
+    table.add_column("№", style="cyan")
+    table.add_column("Содержимое", style="magenta")
+
     for i, doc in enumerate(results, 1):
-        print(f"Результат {i}: {doc.page_content[:200]}...")
+        snippet = (doc.page_content[:200] + "...") if len(doc.page_content) > 200 else doc.page_content
+        table.add_row(str(i), snippet)
+
+    console.print(table)