tcsenpai/multi1
1
0
Fork 0
mirror of https://github.com/tcsenpai/multi1.git synced 2025-06-08 12:05:21 +00:00
Code Issues 4 Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
tikendraw 2024-09-17 22:03:35 +05:30
parent 0c032babe3
commit 3a22b9a81e
1 changed files with 14 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
app/main.py
View File

@ -1,8 +1,9 @@
from time import sleep
import streamlit as st import streamlit as st
from dotenv import load_dotenv from dotenv import load_dotenv
from api_handlers import OllamaHandler, PerplexityHandler, GroqHandler from api_handlers import LitellmHandler
from utils import generate_response from utils import generate_response
from config_menu import config_menu, display_config from config_menu import config_menu
from logger import logger from logger import logger
import os import os
@ -28,13 +29,8 @@ def setup_page():
</p> </p>
""", unsafe_allow_html=True) """, unsafe_allow_html=True)
def get_api_handler(backend, config): def get_api_handler(model, api_key):
if backend == "Ollama": return LitellmHandler(model=model, api_key=api_key)
return OllamaHandler(config['OLLAMA_URL'], config['OLLAMA_MODEL'])
elif backend == "Perplexity AI":
return PerplexityHandler(config['PERPLEXITY_API_KEY'], config['PERPLEXITY_MODEL'])
else: # Groq
return GroqHandler(config['GROQ_API_KEY'], config['GROQ_MODEL'])
def main(): def main():
logger.info("Starting the application") logger.info("Starting the application")
@ -43,10 +39,7 @@ def main():
st.sidebar.markdown('<h3 class="sidebar-title">⚙️ Settings</h3>', unsafe_allow_html=True) st.sidebar.markdown('<h3 class="sidebar-title">⚙️ Settings</h3>', unsafe_allow_html=True)
config = config_menu() config = config_menu()
backend = st.sidebar.selectbox("Choose AI Backend", ["Ollama", "Perplexity AI", "Groq"]) api_handler = get_api_handler(model=config.model, api_key=config.api_key)
display_config(backend, config)
api_handler = get_api_handler(backend, config)
logger.info(f"Selected backend: {backend}")
user_query = st.text_input("💬 Enter your query:", placeholder="e.g., How many 'R's are in the word strawberry?") user_query = st.text_input("💬 Enter your query:", placeholder="e.g., How many 'R's are in the word strawberry?")
@ -56,8 +49,15 @@ def main():
response_container = st.empty() response_container = st.empty()
time_container = st.empty() time_container = st.empty()
max_steps= config.max_steps
max_tokens= config.max_tokens
temperature= config.temperature
timeout= config.timeout
sleeptime= config.sleeptime
try: try:
for steps, total_thinking_time in generate_response(user_query, api_handler): for steps, total_thinking_time in generate_response(user_query, api_handler, max_steps=max_steps, max_tokens=max_tokens, temperature=temperature, timeout=timeout, sleeptime=sleeptime):
with response_container.container(): with response_container.container():
for title, content, _ in steps: for title, content, _ in steps:
if title.startswith("Final Answer"): if title.startswith("Final Answer"):