vietexob commited on
Commit
34a29fb
·
1 Parent(s): ccf9ca7

Renamed HF env vars

Browse files
Files changed (1) hide show
  1. llm_graph.py +4 -4
llm_graph.py CHANGED
@@ -16,8 +16,8 @@ from lightrag.kg.shared_storage import initialize_pipeline_status
16
  load_dotenv()
17
 
18
  # Load the environment variables
19
- api_token = os.environ["HF_TOKEN"]
20
- endpoint_url = os.environ["HF_API_ENDPOINT"]
21
 
22
  AZURE_OPENAI_API_VERSION = os.environ["AZURE_OPENAI_API_VERSION"]
23
  AZURE_OPENAI_DEPLOYMENT = os.environ["AZURE_OPENAI_DEPLOYMENT"]
@@ -81,8 +81,8 @@ class LLMGraph:
81
 
82
  # Hugging Face Inference API for Phi-3-mini-128k-instruct-graph
83
  self.hf_client = InferenceClient(
84
- model=endpoint_url,
85
- token=api_token
86
  )
87
 
88
  self.rag = None # Lazy loading of RAG instance
 
16
  load_dotenv()
17
 
18
  # Load the environment variables
19
+ HF_API_TOKEN = os.environ["HF_TOKEN"]
20
+ HF_API_ENDPOINT = os.environ["HF_API_ENDPOINT"]
21
 
22
  AZURE_OPENAI_API_VERSION = os.environ["AZURE_OPENAI_API_VERSION"]
23
  AZURE_OPENAI_DEPLOYMENT = os.environ["AZURE_OPENAI_DEPLOYMENT"]
 
81
 
82
  # Hugging Face Inference API for Phi-3-mini-128k-instruct-graph
83
  self.hf_client = InferenceClient(
84
+ model=HF_API_ENDPOINT,
85
+ token=HF_API_TOKEN
86
  )
87
 
88
  self.rag = None # Lazy loading of RAG instance