refactor: summarize error message without line number

This commit is contained in:
Florian Hönicke
2023-05-28 10:31:36 +02:00
parent 8333346964
commit 67f2b1e339
2 changed files with 34 additions and 8 deletions

View File

@@ -19,21 +19,26 @@ def answer_yes_no_question(text, question):
question=question, question=question,
text=text, text=text,
) )
# count words "yes" and "no" in pros and cons
yes_count = pros_and_cons.lower().count('yes')
no_count = pros_and_cons.lower().count('no')
return yes_count > no_count
return ask_gpt( # return ask_gpt(
question_prompt, # question_prompt,
boolean_parser, # boolean_parser,
text=text, # text=text,
question=question, # question=question,
pros_and_cons=pros_and_cons, # pros_and_cons=pros_and_cons,
) # )
pros_and_cons_prompt = '''\ pros_and_cons_prompt = '''\
# Context # Context
{text} {text}
# Question # Question
{question} {question}
Note: You must not answer the question. Instead, give up to 5 bullet points (10 words) arguing why the question should be answered with true or false.''' Note: You must not answer the question. Instead, give up to 5 bullet points (10 words) arguing why the question should be answered with yes or no. \
Each bullet point ends with either "- yes" or "- no".'''
question_prompt = '''\ question_prompt = '''\
# Context # Context

View File

@@ -0,0 +1,21 @@
import os
from dev_gpt.apis.gpt import GPTSession
from dev_gpt.options.generate.chains.question_answering import answer_yes_no_question
def test_answer_yes_no_question(tmpdir):
os.environ['VERBOSE'] = 'true'
GPTSession(os.path.join(str(tmpdir), 'log.json'), model='gpt-3.5-turbo')
assert answer_yes_no_question(
'''\
Microservice description:
```
The microservice takes a stock symbol as input and returns a summary of the company's stock performance over the past 30 days, \
including the average closing price and the company name. \
The summary is returned as a string. \
The microservice uses the Yahoo Finance API to fetch the stock data and Python libraries to calculate the average closing price and generate the summary. \
The request parameter is "stock_symbol" and the response parameter is "summary".
```
''', 'Does the microservice send requests to an API beside the google_custom_search and gpt_3_5_turbo?'
) == True