chore: linting (#230)

This commit is contained in:
cachho
2023-07-11 14:04:57 +02:00
committed by GitHub
parent 1c4f629fc0
commit 9409e5605a
3 changed files with 25 additions and 23 deletions

View File

@@ -301,13 +301,13 @@ class App(EmbedChain):
"""
if config is None:
config = InitConfig()
if not config.ef:
config._set_embedding_function_to_default()
if not config.db:
config._set_db_to_default()
super().__init__(config)
def get_llm_model_answer(self, prompt, config: ChatConfig):
@@ -357,12 +357,13 @@ class OpenSourceApp(EmbedChain):
) # noqa:E501
if not config:
config = InitConfig()
if not config.ef:
config._set_embedding_function(
embedding_functions.SentenceTransformerEmbeddingFunction(
model_name="all-MiniLM-L6-v2"
))
embedding_functions.SentenceTransformerEmbeddingFunction(
model_name="all-MiniLM-L6-v2"
)
)
if not config.db:
config._set_db_to_default()