mirror of
https://github.com/microsoft/graphrag.git
synced 2025-03-11 01:26:14 +03:00
e40476153d
* Move verb tests to regular CI * Clean up env vars * Update smoke runtime expectations * Rework artifact assertions * Fix plural in name * remove redundant artifact len check * Remove redundant artifact len check * Adjust graph output expectations * Update community expectations * Include all workflow output * Adjust text unit expectations * Adjust assertions per dataset * Fix test config param name * Update nan allowed for optional model fields --------- Co-authored-by: Alonso Guevara <alonsog@microsoft.com>
45 lines
1.2 KiB
YAML
45 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
|
|
file_pattern: ".*\\.csv$$"
|
|
|
|
snapshots:
|
|
embeddings: True
|
|
|
|
drift_search:
|
|
n_depth: 1
|
|
drift_k_followups: 3
|
|
primer_folds: 3 |