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

utils update

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

18
app/utils.py
View File

@ -1,9 +1,10 @@
import json import json
import time import time
import os import os
from api_handlers import BaseHandler
def generate_response(prompt, api_handler): def generate_response(prompt:str, api_handler:BaseHandler, max_steps:int=5, max_tokens:int=512, temperature:float=0.2, timeout:float = 30.0, sleeptime:float=2.0):
messages = [ messages = [
{ {
"role": "system", "role": "system",
@ -20,9 +21,11 @@ def generate_response(prompt, api_handler):
step_count = 1 step_count = 1
total_thinking_time = 0 total_thinking_time = 0
while True: for _ in range(max_steps):
time.sleep(sleeptime) # to avoid too many requests error
start_time = time.time() start_time = time.time()
step_data = api_handler.make_api_call(messages, 300) step_data = api_handler.make_api_call(messages, max_tokens=max_tokens, temperature=temperature, timeout=timeout)
print(step_data)
end_time = time.time() end_time = time.time()
thinking_time = end_time - start_time thinking_time = end_time - start_time
total_thinking_time += thinking_time total_thinking_time += thinking_time
@ -64,10 +67,7 @@ def generate_response(prompt, api_handler):
def load_env_vars(): def load_env_vars():
return { return {
"OLLAMA_URL": os.getenv("OLLAMA_URL", "http://localhost:11434"), "MODEL": os.getenv("MODEL", "gemini/gemini-1.5-pro"),
"OLLAMA_MODEL": os.getenv("OLLAMA_MODEL", "llama3.1:70b"), "MODEL_API_KEY": os.getenv("MODEL_API_KEY"),
"PERPLEXITY_API_KEY": os.getenv("PERPLEXITY_API_KEY"),
"PERPLEXITY_MODEL": os.getenv(
"PERPLEXITY_MODEL", "llama-3.1-sonar-small-128k-online"
),
} }