mirror of
https://github.com/microsoft/graphrag.git
synced 2026-01-14 09:07:20 +08:00
50 lines
1.3 KiB
YAML
50 lines
1.3 KiB
YAML
models:
|
|
default_chat_model:
|
|
azure_auth_type: api_key
|
|
type: ${GRAPHRAG_LLM_TYPE}
|
|
api_key: ${GRAPHRAG_API_KEY}
|
|
api_base: ${GRAPHRAG_API_BASE}
|
|
api_version: ${GRAPHRAG_API_VERSION}
|
|
deployment_name: ${GRAPHRAG_LLM_DEPLOYMENT_NAME}
|
|
model: ${GRAPHRAG_LLM_MODEL}
|
|
tokens_per_minute: ${GRAPHRAG_LLM_TPM}
|
|
requests_per_minute: ${GRAPHRAG_LLM_RPM}
|
|
model_supports_json: true
|
|
concurrent_requests: 50
|
|
async_mode: threaded
|
|
default_embedding_model:
|
|
azure_auth_type: api_key
|
|
type: ${GRAPHRAG_EMBEDDING_TYPE}
|
|
api_key: ${GRAPHRAG_API_KEY}
|
|
api_base: ${GRAPHRAG_API_BASE}
|
|
api_version: ${GRAPHRAG_API_VERSION}
|
|
deployment_name: ${GRAPHRAG_EMBEDDING_DEPLOYMENT_NAME}
|
|
model: ${GRAPHRAG_EMBEDDING_MODEL}
|
|
tokens_per_minute: ${GRAPHRAG_EMBEDDING_TPM}
|
|
requests_per_minute: ${GRAPHRAG_EMBEDDING_RPM}
|
|
concurrent_requests: 50
|
|
async_mode: threaded
|
|
|
|
vector_store:
|
|
default_vector_store:
|
|
type: "azure_ai_search"
|
|
url: ${AZURE_AI_SEARCH_URL_ENDPOINT}
|
|
api_key: ${AZURE_AI_SEARCH_API_KEY}
|
|
container_name: "simple_text_ci"
|
|
|
|
extract_claims:
|
|
enabled: true
|
|
|
|
community_reports:
|
|
prompt: "prompts/community_report.txt"
|
|
max_length: 2000
|
|
max_input_length: 8000
|
|
|
|
snapshots:
|
|
embeddings: True
|
|
|
|
drift_search:
|
|
n_depth: 1
|
|
k_follow_ups: 3
|
|
primer_folds: 3
|