diego2317 commited on
Commit
2ee6da5
·
1 Parent(s): 7af3c4f

more changes

Browse files
Files changed (2) hide show
  1. chatbot.py +1 -1
  2. mistral_api.py +7 -6
chatbot.py CHANGED
@@ -68,7 +68,7 @@ def answer_query_with_context(
68
  {"role": "system", "content": META_PROMPT},
69
  {"role": "user", "content": prompt}
70
  ]
71
- response = get_response(messages)
72
 
73
  # Append additional service-related information
74
  end_message = (
 
68
  {"role": "system", "content": META_PROMPT},
69
  {"role": "user", "content": prompt}
70
  ]
71
+ response = get_response(message=prompt, system_message=META_PROMPT)
72
 
73
  # Append additional service-related information
74
  end_message = (
mistral_api.py CHANGED
@@ -25,15 +25,16 @@ def get_embedding(text: str, model: str=EMBEDDING_MODEL) -> list[float]:
25
 
26
 
27
 
28
- def get_response(message, history: list[tuple[str, str]], system_message, max_tokens=800, temperature=0) -> str:
29
  """Chat completion using HuggingFace's Inference API"""
30
  messages = [{"role":"system", "content":system_message}]
31
 
32
- for val in history:
33
- if val[0]:
34
- messages.append({"role":"user", "content":val[0]})
35
- if val[1]:
36
- messages.append({"role": "assistant", "content": val[1]})
 
37
 
38
  messages.append({"role": "user", "content": message})
39
 
 
25
 
26
 
27
 
28
+ def get_response(message, history: list[tuple[str, str]]=[[]], system_message="You are a helpful assistant.", max_tokens=800, temperature=0) -> str:
29
  """Chat completion using HuggingFace's Inference API"""
30
  messages = [{"role":"system", "content":system_message}]
31
 
32
+ if history:
33
+ for val in history:
34
+ if val[0]:
35
+ messages.append({"role":"user", "content":val[0]})
36
+ if val[1]:
37
+ messages.append({"role": "assistant", "content": val[1]})
38
 
39
  messages.append({"role": "user", "content": message})
40