mirror of https://github.com/xtekky/gpt4free.git
Merge pull request #1762 from hlohaus/goo
Fix attr conversation_id not found
This commit is contained in:
commit
64e07b7fbf
|
@ -389,19 +389,17 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin):
|
|||
print(f"{e.__class__.__name__}: {e}")
|
||||
|
||||
model = cls.get_model(model).replace("gpt-3.5-turbo", "text-davinci-002-render-sha")
|
||||
fields = Conversation() if conversation is None else copy(conversation)
|
||||
fields = Conversation(conversation_id, parent_id) if conversation is None else copy(conversation)
|
||||
fields.finish_reason = None
|
||||
while fields.finish_reason is None:
|
||||
conversation_id = conversation_id if fields.conversation_id is None else fields.conversation_id
|
||||
parent_id = parent_id if fields.message_id is None else fields.message_id
|
||||
websocket_request_id = str(uuid.uuid4())
|
||||
data = {
|
||||
"action": action,
|
||||
"conversation_mode": {"kind": "primary_assistant"},
|
||||
"force_paragen": False,
|
||||
"force_rate_limit": False,
|
||||
"conversation_id": conversation_id,
|
||||
"parent_message_id": parent_id,
|
||||
"conversation_id": fields.conversation_id,
|
||||
"parent_message_id": fields.message_id,
|
||||
"model": model,
|
||||
"history_and_training_disabled": history_disabled and not auto_continue and not return_conversation,
|
||||
"websocket_request_id": websocket_request_id
|
||||
|
@ -425,6 +423,7 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin):
|
|||
await raise_for_status(response)
|
||||
async for chunk in cls.iter_messages_chunk(response.iter_lines(), session, fields):
|
||||
if return_conversation:
|
||||
history_disabled = False
|
||||
return_conversation = False
|
||||
yield fields
|
||||
yield chunk
|
||||
|
@ -432,7 +431,7 @@ class OpenaiChat(AsyncGeneratorProvider, ProviderModelMixin):
|
|||
break
|
||||
action = "continue"
|
||||
await asyncio.sleep(5)
|
||||
if history_disabled and auto_continue and not return_conversation:
|
||||
if history_disabled and auto_continue:
|
||||
await cls.delete_conversation(session, cls._headers, fields.conversation_id)
|
||||
|
||||
@staticmethod
|
||||
|
|
|
@ -41,7 +41,7 @@ from g4f.providers.base_provider import ProviderModelMixin
|
|||
from g4f.Provider.bing.create_images import patch_provider
|
||||
from g4f.providers.conversation import BaseConversation
|
||||
|
||||
conversations: dict[str, BaseConversation] = {}
|
||||
conversations: dict[dict[str, BaseConversation]] = {}
|
||||
|
||||
class Api():
|
||||
|
||||
|
@ -106,7 +106,8 @@ class Api():
|
|||
kwargs["image"] = open(self.image, "rb")
|
||||
for message in self._create_response_stream(
|
||||
self._prepare_conversation_kwargs(options, kwargs),
|
||||
options.get("conversation_id")
|
||||
options.get("conversation_id"),
|
||||
options.get('provider')
|
||||
):
|
||||
if not window.evaluate_js(f"if (!this.abort) this.add_message_chunk({json.dumps(message)}); !this.abort && !this.error;"):
|
||||
break
|
||||
|
@ -193,8 +194,8 @@ class Api():
|
|||
messages[-1]["content"] = get_search_message(messages[-1]["content"])
|
||||
|
||||
conversation_id = json_data.get("conversation_id")
|
||||
if conversation_id and conversation_id in conversations:
|
||||
kwargs["conversation"] = conversations[conversation_id]
|
||||
if conversation_id and provider in conversations and conversation_id in conversations[provider]:
|
||||
kwargs["conversation"] = conversations[provider][conversation_id]
|
||||
|
||||
model = json_data.get('model')
|
||||
model = model if model else models.default
|
||||
|
@ -211,7 +212,7 @@ class Api():
|
|||
**kwargs
|
||||
}
|
||||
|
||||
def _create_response_stream(self, kwargs, conversation_id: str) -> Iterator:
|
||||
def _create_response_stream(self, kwargs: dict, conversation_id: str, provider: str) -> Iterator:
|
||||
"""
|
||||
Creates and returns a streaming response for the conversation.
|
||||
|
||||
|
@ -231,7 +232,9 @@ class Api():
|
|||
first = False
|
||||
yield self._format_json("provider", get_last_provider(True))
|
||||
if isinstance(chunk, BaseConversation):
|
||||
conversations[conversation_id] = chunk
|
||||
if provider not in conversations:
|
||||
conversations[provider] = {}
|
||||
conversations[provider][conversation_id] = chunk
|
||||
yield self._format_json("conversation", conversation_id)
|
||||
elif isinstance(chunk, Exception):
|
||||
logging.exception(chunk)
|
||||
|
|
|
@ -85,7 +85,7 @@ class Backend_Api(Api):
|
|||
kwargs = self._prepare_conversation_kwargs(json_data, kwargs)
|
||||
|
||||
return self.app.response_class(
|
||||
self._create_response_stream(kwargs, json_data.get("conversation_id")),
|
||||
self._create_response_stream(kwargs, json_data.get("conversation_id"), json_data.get("provider")),
|
||||
mimetype='text/event-stream'
|
||||
)
|
||||
|
||||
|
|
|
@ -20,4 +20,4 @@ aiohttp_socks
|
|||
gpt4all
|
||||
pywebview
|
||||
plyer
|
||||
pycryptodome
|
||||
cryptography
|
24
setup.py
24
setup.py
|
@ -20,22 +20,27 @@ EXTRA_REQUIRE = {
|
|||
"curl_cffi>=0.6.2",
|
||||
"certifi",
|
||||
"browser_cookie3", # get_cookies
|
||||
"PyExecJS", # GptForLove
|
||||
"PyExecJS", # GptForLove, Vercel
|
||||
"duckduckgo-search>=5.0" ,# internet.search
|
||||
"beautifulsoup4", # internet.search and bing.create_images
|
||||
"brotli", # openai
|
||||
#"undetected-chromedriver>=3.5.5", # webdriver
|
||||
#"setuptools", # webdriver
|
||||
"brotli", # openai, bing
|
||||
# webdriver
|
||||
#"undetected-chromedriver>=3.5.5",
|
||||
#"setuptools",
|
||||
#"selenium-wire"
|
||||
# webview
|
||||
"pywebview",
|
||||
"platformdirs",
|
||||
"plyer",
|
||||
"cryptography",
|
||||
####
|
||||
"aiohttp_socks", # proxy
|
||||
"pillow", # image
|
||||
"cairosvg", # svg image
|
||||
"werkzeug", "flask", # gui
|
||||
"loguru", "fastapi",
|
||||
"loguru", "fastapi", # api
|
||||
"uvicorn", "nest_asyncio", # api
|
||||
#"selenium-wire"
|
||||
"pycryptodome" # openai
|
||||
],
|
||||
"image": [
|
||||
"pillow",
|
||||
|
@ -51,9 +56,12 @@ EXTRA_REQUIRE = {
|
|||
"webview": [
|
||||
"webview",
|
||||
"platformdirs",
|
||||
"plyer"
|
||||
"plyer",
|
||||
"cryptography"
|
||||
],
|
||||
"openai": [
|
||||
"pycryptodome"
|
||||
],
|
||||
"openai": [],
|
||||
"api": [
|
||||
"loguru", "fastapi",
|
||||
"uvicorn", "nest_asyncio"
|
||||
|
|
Loading…
Reference in New Issue