Spaces:
Sleeping
Sleeping
Update models/llamaCustom.py
Browse files- models/llamaCustom.py +4 -4
models/llamaCustom.py
CHANGED
@@ -183,10 +183,10 @@ class LlamaCustom:
|
|
183 |
|
184 |
def get_response(self, query_str):
|
185 |
print("query_str: ", query_str)
|
186 |
-
|
187 |
-
query_engine = self.vector_index.as_query_engine(
|
188 |
-
|
189 |
-
)
|
190 |
response = query_engine.query(query_str)
|
191 |
print("metadata: ", response.metadata)
|
192 |
return str(response)
|
|
|
183 |
|
184 |
def get_response(self, query_str):
|
185 |
print("query_str: ", query_str)
|
186 |
+
query_engine = self.vector_index.as_query_engine()
|
187 |
+
# query_engine = self.vector_index.as_query_engine(
|
188 |
+
# text_qa_template=text_qa_template, refine_template=refine_template
|
189 |
+
# )
|
190 |
response = query_engine.query(query_str)
|
191 |
print("metadata: ", response.metadata)
|
192 |
return str(response)
|