optimize code
This commit is contained in:
@@ -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 = {
|
||||
|
||||
@@ -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]
|
||||
|
||||
Reference in New Issue
Block a user