# Configure knowledge graph search
vector_search_settings = {
"use_vector_search": True,
"filters": {"document_type": "article"},
"search_limit": 20,
"use_hybrid_search": True,
"selected_collection_ids": ["c3291abf-8a4e-5d9d-80fd-232ef6fd8526"]
}
# Configure graphRAG search
kg_search_settings = {
"use_kg_search": True,
"kg_search_type": "global",
"kg_search_level": None,
"generation_config": {
"model": "gpt-4",
"temperature": 0.1
},
"entity_types": ["Person", "Organization"],
"relationships": ["worksFor", "foundedBy"],
"max_community_description_length": 65536,
"max_llm_queries_for_global_search": 250,
"local_search_limits": {"__Entity__": 20, "__Relationship__": 20, "__Community__": 20}
}
# Configure LLM generation
rag_generation_config = {
"model": "anthropic/claude-3-opus-20240229",
"temperature": 0.7,
"top_p": 0.95,
"max_tokens_to_sample": 1500,
"stream": True,
"functions": None, # For function calling, if supported
"tools": None, # For tool use, if supported
"add_generation_kwargs": {}, # Additional provider-specific parameters
"api_base": None # Custom API endpoint, if needed
}