Fix embeddings smoke test

AlonsoGuevara · AlonsoGuevara · commit 535c97c0b34c · 2025-02-19T14:15:52.000-06:00
diff --git a/tests/verbs/test_generate_text_embeddings.py b/tests/verbs/test_generate_text_embeddings.py
@@ -7,6 +7,7 @@
     all_embeddings,
 )
 from graphrag.config.enums import LLMType, TextEmbeddingTarget
+from graphrag.index.operations.embed_text.embed_text import TextEmbedStrategyType
 from graphrag.index.workflows.generate_text_embeddings import (
     run_workflow,
 )
@@ -36,6 +37,7 @@ async def test_generate_text_embeddings():
     llm_settings["type"] = LLMType.MockEmbedding
 
     config.embed_text.strategy = {
+        "type": TextEmbedStrategyType.openai,
         "llm": llm_settings,
     }
     config.embed_text.target = TextEmbeddingTarget.all

Original file line number	Diff line number	Diff line change
`@@ -7,6 +7,7 @@`
`7`	`7`	`all_embeddings,`
`8`	`8`	`)`
`9`	`9`	`from graphrag.config.enums import LLMType, TextEmbeddingTarget`
	`10`	`+from graphrag.index.operations.embed_text.embed_text import TextEmbedStrategyType`
`10`	`11`	`from graphrag.index.workflows.generate_text_embeddings import (`
`11`	`12`	`run_workflow,`
`12`	`13`	`)`
`@@ -36,6 +37,7 @@ async def test_generate_text_embeddings():`
`36`	`37`	`llm_settings["type"] = LLMType.MockEmbedding`
`37`	`38`
`38`	`39`	`config.embed_text.strategy = {`
	`40`	`+ "type": TextEmbedStrategyType.openai,`
`39`	`41`	`"llm": llm_settings,`
`40`	`42`	`}`
`41`	`43`	`config.embed_text.target = TextEmbeddingTarget.all`