ivnban27-ctl commited on
Commit
655ffb4
1 Parent(s): 0f381ca

Delete utils.py

Browse files
Files changed (1) hide show
  1. utils.py +0 -34
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)