bupa1018 commited on
Commit
f356efb
·
1 Parent(s): 194ee6d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -8,7 +8,7 @@ from dotenv import load_dotenv
8
  from llm import get_groq_llm
9
  from vectorstore import get_chroma_vectorstore
10
  from embeddings import get_SFR_Code_embedding_model
11
- from kadi_apy_bot import KadiAPYBot
12
 
13
  # Load environment variables from .env file
14
  load_dotenv()
@@ -29,19 +29,20 @@ LLM_MODEL_NAME = config["llm_model_name"]
29
  LLM_MODEL_TEMPERATURE = float(config["llm_model_temperature"])
30
 
31
  def initialize():
32
- global kadiAPY_bot
33
 
34
  vectorstore = get_chroma_vectorstore(get_SFR_Code_embedding_model(), vectorstore_path)
35
  llm = get_groq_llm(LLM_MODEL_NAME, LLM_MODEL_TEMPERATURE, GROQ_API_KEY)
36
 
37
- kadiAPY_bot = KadiAPYBot(llm, vectorstore)
38
 
39
  initialize()
40
 
41
 
 
42
  def bot_kadi(history):
43
  user_query = history[-1][0]
44
- response = kadiAPY_bot.process_query(user_query)
45
  history[-1] = (user_query, response)
46
 
47
  yield history
 
8
  from llm import get_groq_llm
9
  from vectorstore import get_chroma_vectorstore
10
  from embeddings import get_SFR_Code_embedding_model
11
+ from kadiApy_ragchain import KadiApyRagchain
12
 
13
  # Load environment variables from .env file
14
  load_dotenv()
 
29
  LLM_MODEL_TEMPERATURE = float(config["llm_model_temperature"])
30
 
31
  def initialize():
32
+ global kadiAPY_ragchain
33
 
34
  vectorstore = get_chroma_vectorstore(get_SFR_Code_embedding_model(), vectorstore_path)
35
  llm = get_groq_llm(LLM_MODEL_NAME, LLM_MODEL_TEMPERATURE, GROQ_API_KEY)
36
 
37
+ kadiAPY_ragchain = KadiApyRagchain(llm, vectorstore)
38
 
39
  initialize()
40
 
41
 
42
+
43
  def bot_kadi(history):
44
  user_query = history[-1][0]
45
+ response = kadiAPY_ragchain.process_query(user_query)
46
  history[-1] = (user_query, response)
47
 
48
  yield history