From 5a6c809807b7be3d04ed9df6fe518c99d28c0eb6 Mon Sep 17 00:00:00 2001 From: if Date: Wed, 29 Mar 2023 20:37:10 +0300 Subject: [PATCH] fix err handling --- src/utils/{ai_interaction.py => agi/chat_gpt.py} | 1 + src/utils/conversation.py | 8 +++++--- 2 files changed, 6 insertions(+), 3 deletions(-) rename src/utils/{ai_interaction.py => agi/chat_gpt.py} (94%) diff --git a/src/utils/ai_interaction.py b/src/utils/agi/chat_gpt.py similarity index 94% rename from src/utils/ai_interaction.py rename to src/utils/agi/chat_gpt.py index 0ba9b87..85545bd 100644 --- a/src/utils/ai_interaction.py +++ b/src/utils/agi/chat_gpt.py @@ -13,6 +13,7 @@ def send_ai_request(ai_model: str, messages: List[dict]) -> dict: completion = openai.ChatCompletion.create( model=ai_model, messages=messages, + # temperature=0.7, ) logging.warning("completion:") logging.warning(completion) diff --git a/src/utils/conversation.py b/src/utils/conversation.py index 08b075e..b1c6130 100644 --- a/src/utils/conversation.py +++ b/src/utils/conversation.py @@ -2,7 +2,7 @@ import streamlit as st from openai.error import InvalidRequestError, OpenAIError from streamlit_chat import message -from src.utils.ai_interaction import send_ai_request +from src.utils.agi.chat_gpt import send_ai_request from src.utils.tts import show_player @@ -58,11 +58,13 @@ def show_conversation(user_content: str, model: str, role: str) -> None: show_chat(ai_content, user_content) st.markdown("---") show_player(ai_content) - except InvalidRequestError as e: - if e.code == "context_length_exceeded": + except InvalidRequestError as err: + if err.code == "context_length_exceeded": st.session_state.messages.pop(1) if len(st.session_state.messages) == 1: st.session_state.user_text = "" show_conversation(st.session_state.user_text, st.session_state.model, st.session_state.role) + else: + st.error(err) except (OpenAIError, UnboundLocalError) as err: st.error(err)