mirror of
https://github.com/microsoft/graphrag.git
synced 2026-02-14 15:04:29 +08:00
66c2cfb3ce
* Add csv loader tests * Add test loader tests * Add json input support * Remove temp path constraint * Reuse loader cose * Semver * Set file pattern automatically based on type, if empty * Remove pattern from smoke test config * Spelling --------- Co-authored-by: Alonso Guevara <alonsog@microsoft.com>
44 lines
1.2 KiB
YAML
44 lines
1.2 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: "lancedb"
|
|
db_uri: "./tests/fixtures/min-csv/lancedb"
|
|
container_name: "lancedb_ci"
|
|
overwrite: True
|
|
|
|
input:
|
|
file_type: csv
|
|
|
|
snapshots:
|
|
embeddings: True
|
|
|
|
drift_search:
|
|
n_depth: 1
|
|
drift_k_followups: 3
|
|
primer_folds: 3 |