Spaces:
Running
Running
Commit
·
655ffb4
1
Parent(s):
0f381ca
Delete utils.py
Browse files
utils.py
DELETED
@@ -1,34 +0,0 @@
|
|
1 |
-
import streamlit as st
|
2 |
-
from langchain.memory import ConversationBufferMemory
|
3 |
-
|
4 |
-
from models.openai.finetuned_models import finetuned_models, get_finetuned_chain
|
5 |
-
from models.openai.role_models import get_role_chain, role_templates
|
6 |
-
|
7 |
-
def add_initial_message(model_name, memory):
|
8 |
-
if "Spanish" in model_name:
|
9 |
-
memory.chat_memory.add_ai_message("Hola necesito ayuda")
|
10 |
-
else:
|
11 |
-
memory.chat_memory.add_ai_message("Hi I need help")
|
12 |
-
|
13 |
-
def clear_memory(memories):
|
14 |
-
for memory in memories:
|
15 |
-
if memory not in st.session_state:
|
16 |
-
st.session_state[memory] = ConversationBufferMemory(ai_prefix='texter', human_prefix='helper')
|
17 |
-
st.session_state[memory].clear()
|
18 |
-
|
19 |
-
def create_memory_add_initial_message(memories, language):
|
20 |
-
for memory in memories:
|
21 |
-
if memory not in st.session_state:
|
22 |
-
st.session_state[memory] = ConversationBufferMemory(ai_prefix='texter', human_prefix='helper')
|
23 |
-
add_initial_message(language, st.session_state[memory])
|
24 |
-
if len(st.session_state[memory].buffer_as_messages) < 1:
|
25 |
-
add_initial_message(language, st.session_state[memory])
|
26 |
-
|
27 |
-
|
28 |
-
def get_chain(issue, language, source, memory, temperature):
|
29 |
-
if source in ("Finetuned OpenAI"):
|
30 |
-
OA_engine = finetuned_models[f"{issue}-{language}"]
|
31 |
-
return get_finetuned_chain(OA_engine, memory, temperature)
|
32 |
-
if source in ('OpenAI GPT3.5'):
|
33 |
-
template = role_templates[f"{issue}-{language}"]
|
34 |
-
return get_role_chain(template, memory, temperature)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|