hoshingakag commited on
Commit
00bd98a
1 Parent(s): 11555d0

Update src/llamaindex_palm.py

Browse files
Files changed (1) hide show
  1. src/llamaindex_palm.py +1 -1
src/llamaindex_palm.py CHANGED
@@ -167,5 +167,5 @@ class LlamaIndexPaLM():
167
  self,
168
  query: str
169
  ) -> str:
170
- response = self.pinecone_index.as_query_engine().query(query)
171
  return response.response
 
167
  self,
168
  query: str
169
  ) -> str:
170
+ response = self.pinecone_index.as_query_engine(similarity_top_k=3,).query(query)
171
  return response.response