TashaSkyUp
40831eaa28
- Add detailed docstrings to AgentNode, BasicRecursionFilterNode, and DocumentChunkRecursionFilterNode for clarity and documentation. - Improve input validation and error handling in agent nodes. - Update .gitignore to cover more environment, config, and IDE files. - Refactor LiteLLMCustomEndpointProvider for custom API endpoints. - Add conditional imports and dummy functions for LightRAG when unavailable. - Update config.yaml to use environment variables for API keys and clarify usage. - Refine requirements.txt to avoid heavy LightRAG dependencies. - Add tests/__init__.py with test suite metadata; remove redundant test files. - Fix CustomDict import and model config handling in litellmnodes.py. - Improve chunk processing and recursion filter logic in agent nodes. - Update utils/__init__.py to include env_config.
11 lines
202 B
Python
Executable File
11 lines
202 B
Python
Executable File
# lightrag_base.py
|
|
import torch
|
|
import numpy as np
|
|
from typing import Dict, List, Union
|
|
|
|
|
|
class AgentBaseNode:
|
|
def __init__(self):
|
|
self.CATEGORY = "ETK/LLLM_Agents"
|
|
|
|
FUNCTION = "handler" |