Gml 1821 graphrag retriever

[parkererickson-tg]

No description provided.

[RobRossmiller-TG]

One update to the async code within the retriever. Otherwise lgtm

[parkererickson-tg]

I tried this but it didn’t play nicely with FastAPI. Get Outlook for iOS<https://aka.ms/o0ukef>

________________________________ From: RobRossmiller-TG ***@***.***> Sent: Monday, August 19, 2024 7:21:51 PM To: tigergraph/CoPilot ***@***.***> Cc: Parker Erickson ***@***.***>; Author ***@***.***> Subject: Re: [tigergraph/CoPilot] Gml 1821 graphrag retriever (PR #263) @RobRossmiller-TG requested changes on this pull request. One update to the async code within the retriever. Otherwise lgtm

________________________________ In copilot/app/supportai/retrievers/GraphRAG.py<#263 (comment)>:

+ loop = asyncio.new_event_loop()

+ asyncio.set_event_loop(loop) + tasks = [self._generate_candidate(question, c) for c in context] + res = loop.run_until_complete(asyncio.gather(*tasks)) + loop.close() + return res + + def retrieve_answer(self, + question: str, + community_level: int, + top_k_answer_candidates: int = 1): + retrieved = self.search(question, community_level) + context = [x["attributes"] for x in retrieved[0]["selected_comms"]] + + with ThreadPoolExecutor() as executor: + res = executor.submit(self.gather_candidates, question, context).result() If we're going to mix async and not, we can just use an asyncio runner<https://docs.python.org/3/library/asyncio-runner.html#running-an-asyncio-program> instead of starting a new thread and running asyncio in that — Reply to this email directly, view it on GitHub<#263 (review)>, or unsubscribe<https://github.com/notifications/unsubscribe-auth/A36AN3Q7G2NFW5AMUK7Q2Z3ZSKDZ7AVCNFSM6AAAAABMYEWX7SVHI2DSMVQWIX3LMV43YUDVNRWFEZLROVSXG5CSMV3GSZLXHMZDENBWG42TCMRWHA>. You are receiving this because you authored the thread.Message ID: ***@***.***>