Spaces:
Build error
Build error
diego2317 commited on
Commit ·
2ee6da5
1
Parent(s): 7af3c4f
more changes
Browse files- chatbot.py +1 -1
- 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(
|
| 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 |
-
|
| 33 |
-
|
| 34 |
-
|
| 35 |
-
|
| 36 |
-
|
|
|
|
| 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 |
|