diff --git a/solutions/69de7223f309a98be0007e09_8__Самописный_поисковый_агент_на_основе_/agent.py b/solutions/69de7223f309a98be0007e09_8__Самописный_поисковый_агент_на_основе_/agent.py index cfccb00..ee7a6b7 100644 --- a/solutions/69de7223f309a98be0007e09_8__Самописный_поисковый_агент_на_основе_/agent.py +++ b/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]}...") \ No newline at end of file + snippet = (doc.page_content[:200] + "...") if len(doc.page_content) > 200 else doc.page_content + table.add_row(str(i), snippet) + + console.print(table) \ No newline at end of file