optimize code

This commit is contained in:
harry
2024-04-09 19:50:39 +08:00
parent 0b07202d42
commit 6bcd7dc317
2 changed files with 26 additions and 25 deletions

View File

@@ -5,9 +5,9 @@ from typing import List
from loguru import logger
from openai import OpenAI
from openai import AzureOpenAI
import google.generativeai as genai
from app.config import config
def _generate_response(prompt: str) -> str:
content = ""
llm_provider = config.app.get("llm_provider", "openai")
@@ -78,6 +78,7 @@ def _generate_response(prompt: str) -> str:
return content.replace("\n", "")
if llm_provider == "gemini":
import google.generativeai as genai
genai.configure(api_key=api_key)
generation_config = {

View File

@@ -143,7 +143,7 @@ def tr(key):
return loc.get("Translation", {}).get(key, key)
with st.expander(tr("Basic Settings"), expanded=True):
with st.expander(tr("Basic Settings"), expanded=False):
config_panels = st.columns(3)
left_config_panel = config_panels[0]
middle_config_panel = config_panels[1]