Skip to content

Commit 8c1aa21

Browse files
committed
Add models to smoke test configs
1 parent afc9f60 commit 8c1aa21

File tree

2 files changed

+54
-0
lines changed

2 files changed

+54
-0
lines changed

tests/fixtures/min-csv/settings.yml

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,30 @@
1+
models:
2+
default_chat_model:
3+
type: ${GRAPHRAG_LLM_TYPE}
4+
api_key: ${GRAPHRAG_API_KEY}
5+
api_base: ${GRAPHRAG_API_BASE}
6+
api_version: ${GRAPHRAG_API_VERSION}
7+
deployment_name: ${GRAPHRAG_LLM_DEPLOYMENT_NAME}
8+
model: ${GRAPHRAG_LLM_MODEL}
9+
tokens_per_minute: ${GRAPHRAG_LLM_TPM}
10+
requests_per_minute: ${GRAPHRAG_LLM_RPM}
11+
model_supports_json: true
12+
parallelization_num_threads: 50
13+
parallelization_stagger: 0.3
14+
async_mode: threaded
15+
default_embedding_model:
16+
type: ${GRAPHRAG_EMBEDDING_TYPE}
17+
api_key: ${GRAPHRAG_API_KEY}
18+
api_base: ${GRAPHRAG_API_BASE}
19+
api_version: ${GRAPHRAG_API_VERSION}
20+
deployment_name: ${GRAPHRAG_EMBEDDING_DEPLOYMENT_NAME}
21+
model: ${GRAPHRAG_EMBEDDING_MODEL}
22+
tokens_per_minute: ${GRAPHRAG_EMBEDDING_TPM}
23+
requests_per_minute: ${GRAPHRAG_EMBEDDING_RPM}
24+
parallelization_num_threads: 50
25+
parallelization_stagger: 0.3
26+
async_mode: threaded
27+
128
input:
229
file_type: csv
330

tests/fixtures/text/settings.yml

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,30 @@
1+
models:
2+
default_chat_model:
3+
type: ${GRAPHRAG_LLM_TYPE}
4+
api_key: ${GRAPHRAG_API_KEY}
5+
api_base: ${GRAPHRAG_API_BASE}
6+
api_version: ${GRAPHRAG_API_VERSION}
7+
deployment_name: ${GRAPHRAG_LLM_DEPLOYMENT_NAME}
8+
model: ${GRAPHRAG_LLM_MODEL}
9+
tokens_per_minute: ${GRAPHRAG_LLM_TPM}
10+
requests_per_minute: ${GRAPHRAG_LLM_RPM}
11+
model_supports_json: true
12+
parallelization_num_threads: 50
13+
parallelization_stagger: 0.3
14+
async_mode: threaded
15+
default_embedding_model:
16+
type: ${GRAPHRAG_EMBEDDING_TYPE}
17+
api_key: ${GRAPHRAG_API_KEY}
18+
api_base: ${GRAPHRAG_API_BASE}
19+
api_version: ${GRAPHRAG_API_VERSION}
20+
deployment_name: ${GRAPHRAG_EMBEDDING_DEPLOYMENT_NAME}
21+
model: ${GRAPHRAG_EMBEDDING_MODEL}
22+
tokens_per_minute: ${GRAPHRAG_EMBEDDING_TPM}
23+
requests_per_minute: ${GRAPHRAG_EMBEDDING_RPM}
24+
parallelization_num_threads: 50
25+
parallelization_stagger: 0.3
26+
async_mode: threaded
27+
128
claim_extraction:
229
enabled: true
330

0 commit comments

Comments
 (0)