Upload folder using huggingface_hub
Browse files- advanced_rag.py +5 -2
advanced_rag.py
CHANGED
|
@@ -26,6 +26,10 @@ import requests
|
|
| 26 |
from pydantic import PrivateAttr
|
| 27 |
import pydantic
|
| 28 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 29 |
print("Pydantic Version: ")
|
| 30 |
print(pydantic.__version__)
|
| 31 |
# Add Mistral imports with fallback handling
|
|
@@ -173,8 +177,7 @@ class ElevatedRagChain:
|
|
| 173 |
mistral_api_key = os.environ.get("MISTRAL_API_KEY")
|
| 174 |
if not mistral_api_key:
|
| 175 |
raise ValueError("Please set the MISTRAL_API_KEY environment variable to use Mistral API.")
|
| 176 |
-
|
| 177 |
-
from typing import Any, Optional, List
|
| 178 |
class MistralLLM(LLM):
|
| 179 |
temperature: float = 0.7
|
| 180 |
top_p: float = 0.95
|
|
|
|
| 26 |
from pydantic import PrivateAttr
|
| 27 |
import pydantic
|
| 28 |
|
| 29 |
+
from langchain.llms.base import LLM
|
| 30 |
+
from typing import Any, Optional, List
|
| 31 |
+
import typing
|
| 32 |
+
|
| 33 |
print("Pydantic Version: ")
|
| 34 |
print(pydantic.__version__)
|
| 35 |
# Add Mistral imports with fallback handling
|
|
|
|
| 177 |
mistral_api_key = os.environ.get("MISTRAL_API_KEY")
|
| 178 |
if not mistral_api_key:
|
| 179 |
raise ValueError("Please set the MISTRAL_API_KEY environment variable to use Mistral API.")
|
| 180 |
+
|
|
|
|
| 181 |
class MistralLLM(LLM):
|
| 182 |
temperature: float = 0.7
|
| 183 |
top_p: float = 0.95
|