shreyab21 commited on
Commit
1bb9524
·
verified ·
1 Parent(s): 9d6d156

Create retrieval.py

Browse files
Files changed (1) hide show
  1. retrieval.py +11 -0
retrieval.py ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # retrieval.py
2
+ def retrieve_context(query: str) -> str:
3
+ # Simulate retrieval until you hook in embeddings
4
+ mock_knowledge = {
5
+ "agentic": "Agentic AI enables systems to autonomously plan and execute tasks.",
6
+ "llama": "Llama models are advanced open-weight LLMs optimized for reasoning."
7
+ }
8
+ for key, val in mock_knowledge.items():
9
+ if key.lower() in query.lower():
10
+ return val
11
+ return "No relevant context found yet — retrieval DB not connected."