File size: 1,231 Bytes
d9e3edb | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 | import os
import sys
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
import json
from pydantic import ValidationError
from src.assistants.assistant_v1 import gemini_rag_assistant
from src.utils.knowledge_base import AgenticRAG
def main():
rag = AgenticRAG(query_value=True)
while True:
query = input("Enter your query: ")
results = rag.query(query_text=query, n_results=10)
print("\nQuery Results:")
print(json.dumps(results, indent=2))
try:
print("gemini start generating answer")
response = gemini_rag_assistant.get_response_gemini(
message=query, context_data=results
)
print("\nAssistant Response:")
print(response)
except ValidationError as e:
print("Validation Error:", e)
return {
"source_id": "validation_error",
"content": str(e),
}
except Exception as e:
print("Internal Server Error:", e)
return {
"source_id": "internal_error",
"content": "Internal Server Error",
}
if __name__ == "__main__":
main()
|