Spaces:
Sleeping
Sleeping
Update services/utils.py
Browse files- services/utils.py +2 -1
services/utils.py
CHANGED
@@ -45,10 +45,11 @@ def strcuture_document_data(raw_text:str)->dict:
|
|
45 |
parser = PydanticOutputParser(pydantic_object=Candidate)
|
46 |
|
47 |
prompt = PromptTemplate(
|
48 |
-
template=
|
49 |
input_variables=["query"],
|
50 |
partial_variables={"format_instructions": parser.get_format_instructions()},
|
51 |
)
|
|
|
52 |
input = prompt.format_prompt(query=doc_query)
|
53 |
with get_openai_callback() as cb:
|
54 |
result = model(input.to_string())
|
|
|
45 |
parser = PydanticOutputParser(pydantic_object=Candidate)
|
46 |
|
47 |
prompt = PromptTemplate(
|
48 |
+
template="Answer the user query.\n{format_instructions}\n{query}\n",
|
49 |
input_variables=["query"],
|
50 |
partial_variables={"format_instructions": parser.get_format_instructions()},
|
51 |
)
|
52 |
+
print(prompt)
|
53 |
input = prompt.format_prompt(query=doc_query)
|
54 |
with get_openai_callback() as cb:
|
55 |
result = model(input.to_string())
|