Feat: optimize error desc (#152)

2023-05-22 17:39:28 +08:00 · 2023-05-22 17:39:28 +08:00 · 9a5ae9f51f
commit 9a5ae9f51f
parent a7c40a07d8
15 changed files with 119 additions and 55 deletions
--- a/api/core/index/index_builder.py
+++ b/api/core/index/index_builder.py
@ -46,3 +46,15 @@ class IndexBuilder:
            prompt_helper=prompt_helper,
            embed_model=OpenAIEmbedding(**model_credentials),
        )
+
+    @classmethod
+    def get_fake_llm_service_context(cls, tenant_id: str) -> ServiceContext:
+        llm = LLMBuilder.to_llm(
+            tenant_id=tenant_id,
+            model_name='fake'
+        )
+
+        return ServiceContext.from_defaults(
+            llm_predictor=LLMPredictor(llm=llm),
+            embed_model=OpenAIEmbedding()
+        )