Spaces:
Running
Running
| import os | |
| import asyncio | |
| from autoagents.agents.search import ActionRunner | |
| from langchain.callbacks import get_openai_callback | |
| from pprint import pprint | |
| import pdb | |
| from ast import literal_eval | |
| from multiprocessing import Pool, TimeoutError | |
| async def work(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 | |
| Q = [ | |
| "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." | |
| "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?" | |
| ] | |
| def main(q): | |
| asyncio.run(work(q)) | |
| if __name__ == "__main__": | |
| with Pool(processes=10) as pool: | |
| print(pool.map(main, Q)) | |