formatting

Author: strickvl

llm-complete-guide/steps/eval_retrieval.py

@@ -97,7 +97,9 @@ def query_similar_docs(
         conn = get_db_conn()
     elif vector_store_name == "pinecone":
         # in pipeline runs, always use staging index
-        pinecone_index = get_pinecone_client(model_version_name_or_id="staging")
+        pinecone_index = get_pinecone_client(
+            model_version_name_or_id="staging"
+        )
     else:
         es_client = get_es_client()
 

llm-complete-guide/steps/populate_index.py

@@ -831,14 +831,18 @@ def _index_generator_postgres(documents: str) -> None:
             conn.close()
 
 
-def _index_generator_pinecone(documents: str, model_version_name_or_id: str) -> None:
+def _index_generator_pinecone(
+    documents: str, model_version_name_or_id: str
+) -> None:
     """Generates a Pinecone index for the given documents.
 
     Args:
         documents (str): JSON string containing the documents to index.
         model_version (str): Name of the model version.
     """
-    index = get_pinecone_client(model_version_name_or_id=model_version_name_or_id)
+    index = get_pinecone_client(
+        model_version_name_or_id=model_version_name_or_id
+    )
 
     # Load documents
     docs = json.loads(documents)