Spaces:
Running
Running
Upload folder using huggingface_hub
Browse files- 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 |
|