graphrag

graphrag.py

@@ -9,21 +9,15 @@ llm = ChatOpenAI(temperature=0, model_name="gpt-4-turbo")
 from langchain_ollama.llms import OllamaLLM
 llm = OllamaLLM(temperature=0,model="llama3.2")
 """
-
-from typing import List, Optional
-
 from llama_index.llms.huggingface_api import HuggingFaceInferenceAPI
 
 # SEE: https://huggingface.co/docs/hub/security-tokens
 # We just need a token with read permissions for this demo
-HF_TOKEN: Optional[str] = os.getenv("HF_TOKEN")
+HF_TOKEN= os.environ["HF_TOKEN"]
+
+llm = HuggingFaceInferenceAPI(temperature=0, model_name="meta-llama/Llama-3.2-1B")
 
-llm = HuggingFaceInferenceAPI(
-    temperature=0,
-    model_name="meta-llama/Llama-3.2-1B"
-)
 
-import os
 import networkx as nx
 import matplotlib.pyplot as plt
 import pandas as pd