fix weaviate embedded client

hemanoid · Nov 29, 2023 · d03901f · d03901f
1 parent 1e08b7f
commit d03901f
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/goldenverba/verba_manager.py b/goldenverba/verba_manager.py
@@ -215,7 +215,7 @@ def setup_client(self) -> Client | None:
             msg.info("Using Weaviate Embedded")
             self.weaviate_type = "Weaviate Embedded"
             client = weaviate.Client(
-                additional_headers={"X-OpenAI-Api-Key": openai.api_key},
+                additional_headers=additional_header,
                 embedded_options=EmbeddedOptions(),
             )
 
@@ -418,7 +418,10 @@ def set_suggestions(self, query: str):
             .do()
         )
 
-        if "data" in check_results and len(check_results["data"]["Get"]["Suggestion"]) > 0:
+        if (
+            "data" in check_results
+            and len(check_results["data"]["Get"]["Suggestion"]) > 0
+        ):
             if query == check_results["data"]["Get"]["Suggestion"][0]["suggestion"]:
                 return