Spaces:
Running
Running
| import os | |
| import asyncio | |
| from action import ActionRunner | |
| from langchain.callbacks import get_openai_callback | |
| from pprint import pprint | |
| import pdb | |
| from ast import literal_eval | |
| async def main(user_input): | |
| outputq = asyncio.Queue() | |
| API_O = os.getenv("OPENAI_API_KEY") | |
| runner = ActionRunner(outputq, api_key=API_O, model_name="gpt-3.5-turbo") | |
| task = asyncio.create_task(runner.run(user_input, outputq)) | |
| while True: | |
| output = await outputq.get() | |
| if isinstance(output, Exception): | |
| print(output) | |
| return | |
| try: | |
| pprint(literal_eval(output)) | |
| except: | |
| print(output) | |
| print("-----------------------------------------------------------") | |
| if "Final Answer:" in output: | |
| break | |
| await task | |
| # "list 5 cities and their current populations where Paramore is playing this year.", | |
| # "Who is Leo DiCaprio's girlfriend? What is her current age raised to the 0.43 power?", | |
| # "How many watermelons can fit in a Tesla Model S?", | |
| # "Recommend me some laptops suitable for UI designers under $2000. Please include brand and price." | |
| Q = [ | |
| "Build me a vacation plan for Rome and Milan this summer for seven days. Include place to visit and hotels to stay. ", | |
| "What is the sum of ages of the wives of Barack Obama and Donald Trump?", | |
| "Who is the most recent NBA MVP? Which team does he play for? What is his season stats?", | |
| "What were the scores for the last three games for the Los Angeles Lakers? Provide the dates and opposing teams.", | |
| "Which team won in women's volleyball in the Summer Olympics that was held in London?", | |
| "Provide a summary of the latest COVID-19 research paper published. Include the title, authors and abstract.", | |
| "What is the top grossing movie in theatres this week? Provide the movie title, director, and a brief synopsis of the movie's plot. Attach a review for this movie.", | |
| "Recommend a bagel shop near the Strip district in Pittsburgh that offer vegan food", | |
| "Who are some top researchers in the field of machine learning systems nowadays?" | |
| ] | |
| loop = asyncio.new_event_loop() | |
| for i in range(len(Q)): | |
| loop.run_until_complete(main(Q[i])) | |