rishabhsetiya commited on
Commit
9b3cf89
·
verified ·
1 Parent(s): e7df182

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -72,7 +72,7 @@ def get_mistral_answer(query: str, context: str) -> str:
72
  try:
73
  logger.info(f"Calling Mistral API for query: {query}")
74
  completion = client.chat.completions.create(
75
- model="mistralai/Mistral-7B-Instruct-v0.2:featherless-ai",
76
  messages=[
77
  {
78
  "role": "user",
 
72
  try:
73
  logger.info(f"Calling Mistral API for query: {query}")
74
  completion = client.chat.completions.create(
75
+ model="dphn/Dolphin-Mistral-24B-Venice-Edition:featherless-ai",
76
  messages=[
77
  {
78
  "role": "user",