temporal-graphrag 1.0.0__tar.gz

temporal_graphrag-1.0.0/PKG-INFO

@@ -0,0 +1,118 @@
+Metadata-Version: 2.4
+Name: temporal-graphrag
+Version: 1.0.0
+Summary: HelixRAG: Retrieval-Augmented Generation with Graphiti-powered Knowledge Graphs
+Author-email: Yash Nuhash <nuhashroxme@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/YashNuhash/Helix
+Project-URL: Documentation, https://github.com/YashNuhash/Helix#readme
+Project-URL: Repository, https://github.com/YashNuhash/Helix
+Project-URL: Bug Tracker, https://github.com/YashNuhash/Helix/issues
+Keywords: rag,graphrag,knowledge-graph,graphiti,llm,ai,neo4j
+Classifier: Development Status :: 4 - Beta
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: aiohttp
+Requires-Dist: configparser
+Requires-Dist: google-api-core<3.0.0,>=2.0.0
+Requires-Dist: google-genai<2.0.0,>=1.0.0
+Requires-Dist: json_repair
+Requires-Dist: nano-vectordb
+Requires-Dist: networkx
+Requires-Dist: numpy<3.0.0,>=1.24.0
+Requires-Dist: pandas<2.4.0,>=2.0.0
+Requires-Dist: pipmaster
+Requires-Dist: pydantic
+Requires-Dist: pypinyin
+Requires-Dist: python-dotenv
+Requires-Dist: setuptools
+Requires-Dist: tenacity
+Requires-Dist: tiktoken
+Requires-Dist: xlsxwriter>=3.1.0
+Requires-Dist: graphiti-core>=0.27.1
+Requires-Dist: neo4j>=5.28.0
+Provides-Extra: pytest
+Requires-Dist: pytest>=8.4.2; extra == "pytest"
+Requires-Dist: pytest-asyncio>=1.2.0; extra == "pytest"
+Requires-Dist: pre-commit; extra == "pytest"
+Requires-Dist: ruff; extra == "pytest"
+Provides-Extra: api
+Requires-Dist: aiohttp; extra == "api"
+Requires-Dist: configparser; extra == "api"
+Requires-Dist: json_repair; extra == "api"
+Requires-Dist: nano-vectordb; extra == "api"
+Requires-Dist: networkx; extra == "api"
+Requires-Dist: numpy<3.0.0,>=1.24.0; extra == "api"
+Requires-Dist: openai<3.0.0,>=2.0.0; extra == "api"
+Requires-Dist: pandas<2.4.0,>=2.0.0; extra == "api"
+Requires-Dist: pipmaster; extra == "api"
+Requires-Dist: pydantic; extra == "api"
+Requires-Dist: pypinyin; extra == "api"
+Requires-Dist: python-dotenv; extra == "api"
+Requires-Dist: setuptools; extra == "api"
+Requires-Dist: tenacity; extra == "api"
+Requires-Dist: tiktoken; extra == "api"
+Requires-Dist: xlsxwriter>=3.1.0; extra == "api"
+Requires-Dist: google-api-core<3.0.0,>=2.0.0; extra == "api"
+Requires-Dist: google-genai<2.0.0,>=1.0.0; extra == "api"
+Requires-Dist: aiofiles; extra == "api"
+Requires-Dist: ascii_colors; extra == "api"
+Requires-Dist: distro; extra == "api"
+Requires-Dist: fastapi; extra == "api"
+Requires-Dist: httpcore; extra == "api"
+Requires-Dist: httpx>=0.28.1; extra == "api"
+Requires-Dist: jiter; extra == "api"
+Requires-Dist: bcrypt>=4.0.0; extra == "api"
+Requires-Dist: psutil; extra == "api"
+Requires-Dist: PyJWT<3.0.0,>=2.8.0; extra == "api"
+Requires-Dist: python-jose[cryptography]; extra == "api"
+Requires-Dist: python-multipart; extra == "api"
+Requires-Dist: pytz; extra == "api"
+Requires-Dist: uvicorn; extra == "api"
+Requires-Dist: gunicorn; extra == "api"
+Requires-Dist: openpyxl<4.0.0,>=3.0.0; extra == "api"
+Requires-Dist: pycryptodome<4.0.0,>=3.0.0; extra == "api"
+Requires-Dist: pypdf>=6.1.0; extra == "api"
+Requires-Dist: python-docx<2.0.0,>=0.8.11; extra == "api"
+Requires-Dist: python-pptx<2.0.0,>=0.6.21; extra == "api"
+Provides-Extra: offline-storage
+Requires-Dist: redis<8.0.0,>=5.0.0; extra == "offline-storage"
+Requires-Dist: neo4j<7.0.0,>=5.0.0; extra == "offline-storage"
+Requires-Dist: pymilvus<3.0.0,>=2.6.2; extra == "offline-storage"
+Requires-Dist: pymongo<5.0.0,>=4.0.0; extra == "offline-storage"
+Requires-Dist: asyncpg<1.0.0,>=0.31.0; extra == "offline-storage"
+Requires-Dist: pgvector<1.0.0,>=0.4.2; extra == "offline-storage"
+Requires-Dist: qdrant-client<2.0.0,>=1.11.0; extra == "offline-storage"
+Provides-Extra: offline-llm
+Requires-Dist: openai<3.0.0,>=2.0.0; extra == "offline-llm"
+Requires-Dist: anthropic<1.0.0,>=0.18.0; extra == "offline-llm"
+Requires-Dist: ollama<1.0.0,>=0.1.0; extra == "offline-llm"
+Requires-Dist: zhipuai<3.0.0,>=2.0.0; extra == "offline-llm"
+Requires-Dist: aioboto3<16.0.0,>=12.0.0; extra == "offline-llm"
+Requires-Dist: voyageai<1.0.0,>=0.2.0; extra == "offline-llm"
+Requires-Dist: llama-index<1.0.0,>=0.14.0; extra == "offline-llm"
+Requires-Dist: llama-index-llms-openai>=0.6.12; extra == "offline-llm"
+Requires-Dist: google-api-core<3.0.0,>=2.0.0; extra == "offline-llm"
+Requires-Dist: google-genai<2.0.0,>=1.0.0; extra == "offline-llm"
+Provides-Extra: offline
+Requires-Dist: helixrag[api,offline-llm,offline-storage]; extra == "offline"
+Provides-Extra: test
+Requires-Dist: helixrag[api]; extra == "test"
+Requires-Dist: pytest>=8.4.2; extra == "test"
+Requires-Dist: pytest-asyncio>=1.2.0; extra == "test"
+Requires-Dist: pre-commit; extra == "test"
+Requires-Dist: ruff; extra == "test"
+Provides-Extra: evaluation
+Requires-Dist: helixrag[api]; extra == "evaluation"
+Requires-Dist: ragas>=0.3.7; extra == "evaluation"
+Requires-Dist: datasets>=4.3.0; extra == "evaluation"

temporal_graphrag-1.0.0/api/__init__.py

@@ -0,0 +1 @@
+__api_version__ = "0271"

temporal_graphrag-1.0.0/api/auth.py

@@ -0,0 +1,109 @@
+from datetime import datetime, timedelta
+
+import jwt
+from dotenv import load_dotenv
+from fastapi import HTTPException, status
+from pydantic import BaseModel
+
+from .config import global_args
+
+# use the .env that is inside the current folder
+# allows to use different .env file for each lightrag instance
+# the OS environment variables take precedence over the .env file
+load_dotenv(dotenv_path=".env", override=False)
+
+
+class TokenPayload(BaseModel):
+    sub: str  # Username
+    exp: datetime  # Expiration time
+    role: str = "user"  # User role, default is regular user
+    metadata: dict = {}  # Additional metadata
+
+
+class AuthHandler:
+    def __init__(self):
+        self.secret = global_args.token_secret
+        self.algorithm = global_args.jwt_algorithm
+        self.expire_hours = global_args.token_expire_hours
+        self.guest_expire_hours = global_args.guest_token_expire_hours
+        self.accounts = {}
+        auth_accounts = global_args.auth_accounts
+        if auth_accounts:
+            for account in auth_accounts.split(","):
+                username, password = account.split(":", 1)
+                self.accounts[username] = password
+
+    def create_token(
+        self,
+        username: str,
+        role: str = "user",
+        custom_expire_hours: int = None,
+        metadata: dict = None,
+    ) -> str:
+        """
+        Create JWT token
+
+        Args:
+            username: Username
+            role: User role, default is "user", guest is "guest"
+            custom_expire_hours: Custom expiration time (hours), if None use default value
+            metadata: Additional metadata
+
+        Returns:
+            str: Encoded JWT token
+        """
+        # Choose default expiration time based on role
+        if custom_expire_hours is None:
+            if role == "guest":
+                expire_hours = self.guest_expire_hours
+            else:
+                expire_hours = self.expire_hours
+        else:
+            expire_hours = custom_expire_hours
+
+        expire = datetime.utcnow() + timedelta(hours=expire_hours)
+
+        # Create payload
+        payload = TokenPayload(
+            sub=username, exp=expire, role=role, metadata=metadata or {}
+        )
+
+        return jwt.encode(payload.dict(), self.secret, algorithm=self.algorithm)
+
+    def validate_token(self, token: str) -> dict:
+        """
+        Validate JWT token
+
+        Args:
+            token: JWT token
+
+        Returns:
+            dict: Dictionary containing user information
+
+        Raises:
+            HTTPException: If token is invalid or expired
+        """
+        try:
+            payload = jwt.decode(token, self.secret, algorithms=[self.algorithm])
+            expire_timestamp = payload["exp"]
+            expire_time = datetime.utcfromtimestamp(expire_timestamp)
+
+            if datetime.utcnow() > expire_time:
+                raise HTTPException(
+                    status_code=status.HTTP_401_UNAUTHORIZED, detail="Token expired"
+                )
+
+            # Return complete payload instead of just username
+            return {
+                "username": payload["sub"],
+                "role": payload.get("role", "user"),
+                "metadata": payload.get("metadata", {}),
+                "exp": expire_time,
+            }
+        except jwt.PyJWTError:
+            raise HTTPException(
+                status_code=status.HTTP_401_UNAUTHORIZED, detail="Invalid token"
+            )
+
+
+auth_handler = AuthHandler()