CaffeinatedCoding commited on
Commit
b2d9591
·
verified ·
1 Parent(s): c585084

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. src/agent_v2.py +6 -0
src/agent_v2.py CHANGED
@@ -378,6 +378,12 @@ def run_query_v2(user_message: str, session_id: str) -> Dict[str, Any]:
378
  chunks = []
379
  try:
380
  chunks = retrieve_parallel(search_queries[:3], top_k=5)
 
 
 
 
 
 
381
  except Exception as e:
382
  logger.error(f"Pass 2 failed: {e}")
383
 
 
378
  chunks = []
379
  try:
380
  chunks = retrieve_parallel(search_queries[:3], top_k=5)
381
+ # Add precedent chain
382
+ from src.citation_graph import get_precedent_chain
383
+ retrieved_ids = [c.get("judgment_id", "") for c in chunks]
384
+ precedents = get_precedent_chain(retrieved_ids, max_precedents=2)
385
+ if precedents:
386
+ chunks.extend(precedents)
387
  except Exception as e:
388
  logger.error(f"Pass 2 failed: {e}")
389