edit index_nodes log to print both models

TheDataGuild · Sep 23, 2023 · 801d303 · 801d303
1 parent 9c255be
commit 801d303
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/mind_palace/index.py b/mind_palace/index.py
@@ -6,13 +6,16 @@
 
 @time_function
 def index_nodes(nodes, model="gpt-3.5-turbo"):
+    embed_model = "local"
     service_context = li.ServiceContext.from_defaults(
         llm=OpenAI(model=model),
-        embed_model="local"
+        embed_model=embed_model
         # Q: how much does a better embedding model help?
     )
 
-    print(f"Creating an index with {len(nodes)} number of nodes using model {model}")
+    print(
+        f"Creating an index with {len(nodes)} number of nodes using embedding model {embed_model} and querying LLM {model}"
+    )
     index = li.VectorStoreIndex(nodes, service_context=service_context)
 
     return index