Spaces:
Sleeping
Sleeping
Update src/agents/rag_agent.py
Browse files- src/agents/rag_agent.py +1 -1
src/agents/rag_agent.py
CHANGED
|
@@ -34,7 +34,7 @@ def search_legal_documents(query: str) -> str:
|
|
| 34 |
"""Search Moroccan **legal codes, decrees, and PDFs**. Use this tool when user asks about Moroccan **laws, articles, codes, legal references**, or related content."""
|
| 35 |
logging.info(f"Tool call: search_legal_documents with input: {query}")
|
| 36 |
try:
|
| 37 |
-
relevant_docs = search_relevant_documents(query, top_k=
|
| 38 |
results = doc_rag_pipeline.get_top_docs_chunks_for_query(query, relevant_docs['results'])
|
| 39 |
output = "\n".join([f"{r['text']}\n(Source: {r['pdf_link']})" for r in results])
|
| 40 |
logging.info(f"Tool result (search_legal_documents): {output}")
|
|
|
|
| 34 |
"""Search Moroccan **legal codes, decrees, and PDFs**. Use this tool when user asks about Moroccan **laws, articles, codes, legal references**, or related content."""
|
| 35 |
logging.info(f"Tool call: search_legal_documents with input: {query}")
|
| 36 |
try:
|
| 37 |
+
relevant_docs = search_relevant_documents(query, top_k=1)
|
| 38 |
results = doc_rag_pipeline.get_top_docs_chunks_for_query(query, relevant_docs['results'])
|
| 39 |
output = "\n".join([f"{r['text']}\n(Source: {r['pdf_link']})" for r in results])
|
| 40 |
logging.info(f"Tool result (search_legal_documents): {output}")
|