Update src/llamaindex_palm.py

src/llamaindex_palm.py

@@ -283,7 +283,9 @@ class LlamaIndexPaLM():
             # No history, single context retrieval without evaluation
             if not history:
                 # w&b trace retrieval context
-                result_query_only = asyncio.run(self.retrieve_context(query))
+                result_query_only = self.retrieve_context(query)
+                # async version
+                # result_query_only = asyncio.run(self.retrieve_context(query))
                 context_from_index_selected = result_query_only["result"]
                 agent_end_time_ms = round(datetime.datetime.now().timestamp() * 1000)
                 retrieval_span = Trace(