alx-d commited on
Commit
01f968f
·
verified ·
1 Parent(s): e02cfbe

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
- from langchain.llms.base import LLM
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