Skip to content

Commit d0a4c33

Browse files
committed
Swtich smoke tests to LiteLLM
1 parent 31e8849 commit d0a4c33

File tree

2 files changed

+8
-4
lines changed

2 files changed

+8
-4
lines changed

tests/fixtures/min-csv/settings.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
models:
22
default_chat_model:
33
azure_auth_type: api_key
4-
type: ${GRAPHRAG_LLM_TYPE}
4+
type: chat
5+
model_provider: azure
56
api_key: ${GRAPHRAG_API_KEY}
67
api_base: ${GRAPHRAG_API_BASE}
78
api_version: ${GRAPHRAG_API_VERSION}
@@ -14,7 +15,8 @@ models:
1415
async_mode: threaded
1516
default_embedding_model:
1617
azure_auth_type: api_key
17-
type: ${GRAPHRAG_EMBEDDING_TYPE}
18+
type: embedding
19+
model_provider: azure
1820
api_key: ${GRAPHRAG_API_KEY}
1921
api_base: ${GRAPHRAG_API_BASE}
2022
api_version: ${GRAPHRAG_API_VERSION}

tests/fixtures/text/settings.yml

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,8 @@
11
models:
22
default_chat_model:
33
azure_auth_type: api_key
4-
type: ${GRAPHRAG_LLM_TYPE}
4+
type: chat
5+
model_provider: azure
56
api_key: ${GRAPHRAG_API_KEY}
67
api_base: ${GRAPHRAG_API_BASE}
78
api_version: ${GRAPHRAG_API_VERSION}
@@ -14,7 +15,8 @@ models:
1415
async_mode: threaded
1516
default_embedding_model:
1617
azure_auth_type: api_key
17-
type: ${GRAPHRAG_EMBEDDING_TYPE}
18+
type: embedding
19+
model_provider: azure
1820
api_key: ${GRAPHRAG_API_KEY}
1921
api_base: ${GRAPHRAG_API_BASE}
2022
api_version: ${GRAPHRAG_API_VERSION}

0 commit comments

Comments
 (0)