Update apps/agents/agents.py

apps/agents/agents.py

@@ -24,14 +24,14 @@ from typing import Any, Dict, List, Optional, Tuple, Union
 
 import gradio as gr
 import openai
-import openai.error
+
 import tenacity
 
 from apps.agents.text_utils import split_markdown_code
 from camel.agents import TaskSpecifyAgent
 from camel.messages import BaseMessage
 from camel.societies import RolePlaying
-from camel.typing import TaskType
+from camel.types import TaskType
 
 REPO_ROOT = os.path.realpath(
     os.path.join(os.path.dirname(os.path.abspath(__file__)), "../.."))
@@ -246,8 +246,8 @@ def role_playing_chat_init(state) -> \
 
     try:
         init_assistant_msg: BaseMessage
-        init_assistant_msg, _ = session.init_chat()
-    except (openai.error.RateLimitError, tenacity.RetryError,
+        init_assistant_msg = session.init_chat()
+    except (openai.RateLimitError, tenacity.RetryError,
             RuntimeError) as ex:
         print("OpenAI API exception 1 " + str(ex))
         state.session = None
@@ -289,7 +289,7 @@ def role_playing_chat_cont(state) -> \
     try:
         assistant_response, user_response = session.step(
             state.saved_assistant_msg)
-    except (openai.error.RateLimitError, tenacity.RetryError,
+    except (openai.RateLimitError, tenacity.RetryError,
             RuntimeError) as ex:
         print("OpenAI API exception 2 " + str(ex))
         state.session = None
@@ -416,8 +416,7 @@ def construct_ui(blocks, api_key: Optional[str] = None) -> None:
                 "Github repo: [https://github.com/lightaime/camel]"
                 "(https://github.com/lightaime/camel)"
                 '<div style="display:flex; justify-content:center;">'
-                '<img src="https://raw.githubusercontent.com/lightaime/camel/'
-                'master/misc/logo.png" alt="Logo" style="max-width:50%;">'
+                '<img src="https://raw.githubusercontent.com/camel-ai/camel/master/misc/primary_logo.png" alt="Logo" style="max-width:50%;">'
                 '</div>')
     with gr.Row():
         with gr.Column(scale=9):