sxid003 commited on
Commit
8af85fa
·
verified ·
1 Parent(s): e6d636d

Update src/agents/rag_agent.py

Browse files
Files changed (1) hide show
  1. 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=3)
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}")