From 6e0bc147b52cb1e52d7fb3f8dd01d1f33dae201e Mon Sep 17 00:00:00 2001 From: Heiner Lohaus Date: Fri, 3 Jan 2025 20:35:46 +0100 Subject: Support continue messages in Airforce Add auth caching for OpenAI ChatGPT Some provider improvments --- g4f/Provider/Copilot.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'g4f/Provider/Copilot.py') diff --git a/g4f/Provider/Copilot.py b/g4f/Provider/Copilot.py index 828ab718..76365da1 100644 --- a/g4f/Provider/Copilot.py +++ b/g4f/Provider/Copilot.py @@ -90,7 +90,7 @@ class Copilot(AbstractProvider, ProviderModelMixin): cls._access_token, cls._cookies = asyncio.run(get_access_token_and_cookies(cls.url, proxy)) else: raise h - yield Parameters(**{"api_key": cls._access_token, "cookies": cls._cookies}) + yield Parameters(**{"api_key": cls._access_token, "cookies": cls._cookies if isinstance(cls._cookies, dict) else {c.name: c.value for c in cls._cookies}}) websocket_url = f"{websocket_url}&accessToken={quote(cls._access_token)}" headers = {"authorization": f"Bearer {cls._access_token}"} @@ -191,6 +191,8 @@ class Copilot(AbstractProvider, ProviderModelMixin): yield ImageResponse(msg.get("url"), image_prompt, {"preview": msg.get("thumbnailUrl")}) elif msg.get("event") == "done": break + elif msg.get("event") == "replaceText": + yield msg.get("text") elif msg.get("event") == "error": raise RuntimeError(f"Error: {msg}") elif msg.get("event") not in ["received", "startMessage", "citation", "partCompleted"]: -- cgit v1.2.3