File size: 756 Bytes
c92680a | 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 | #!/usr/bin/env python3
import sys
sys.path.append("/home/bhaskar/cd/campusguide")
from retriever.retriever import Retriever
def main():
retriever = Retriever()
query = "What documents are students expected to have during placement induction?"
result = retriever.retrieve(query, role="student")
print(f"Retrieved {len(result['chunks'])} chunks:")
print(f"Confidence: {result.get('confidence', 'N/A')}")
print(f"Query: {result.get('query', 'N/A')}")
print(f"Role: {result.get('role', 'N/A')}")
for i, chunk in enumerate(result["chunks"], 1):
print(f"\nChunk {i}:")
print(f"Text: {chunk['text'][:200]}...")
print(f"Metadata: {chunk.get('metadata', {})}")
if __name__ == "__main__":
main()
|