#!/usr/bin/env python3
import sys
import os
sys.path.insert(0, os.path.dirname(__file__))
from code_context_mcp import CodeContextManager
def test_search():
manager = CodeContextManager()
# Re-index files
manager.index_file("code_context_mcp.py")
manager.index_file("test.js")
queries = ["function to parse python files", "class CodeParser", "vector store", "Python file with entities", "javascript function", "class Person"]
for query in queries:
results = manager.get_project_context(query, max_files=5, max_entities=10)
print(f"Search results for query '{query}':")
for i, result in enumerate(results[:10], 1): # limit to 10
print(f"{i}. Type: {result.get('type', 'N/A')}, File: {result.get('file_path', 'N/A')}, Name: {result.get('name', 'N/A')}, Similarity: {result.get('similarity', 'N/A')}")
if 'signature' in result:
print(f" Signature: {result['signature']}")
if 'content' in result:
print(f" Content preview: {result['content'][:100]}...")
print()
if __name__ == "__main__":
test_search()
