Fix load_dotenv() missing problem for base.py

This commit is contained in:
yangdx
2025-02-16 20:05:45 +08:00
parent 36ff0bfa85
commit 4e354451b7

View File

@@ -1,4 +1,5 @@
import os import os
from dotenv import load_dotenv
from dataclasses import dataclass, field from dataclasses import dataclass, field
from enum import Enum from enum import Enum
from typing import ( from typing import (
@@ -9,12 +10,11 @@ from typing import (
TypeVar, TypeVar,
Union, Union,
) )
import numpy as np import numpy as np
from .utils import EmbeddingFunc from .utils import EmbeddingFunc
from .types import KnowledgeGraph from .types import KnowledgeGraph
load_dotenv()
class TextChunkSchema(TypedDict): class TextChunkSchema(TypedDict):
tokens: int tokens: int