Merge branch 'main' into code-cleaning

This commit is contained in:
zrguo
2025-02-16 19:41:05 +08:00
committed by GitHub
13 changed files with 822 additions and 183 deletions

View File

@@ -1,7 +1,7 @@
from __future__ import annotations
from pydantic import BaseModel
from typing import Any
from typing import List, Dict, Any, Optional
class GPTKeywordExtractionFormat(BaseModel):
@@ -17,7 +17,7 @@ class KnowledgeGraphNode(BaseModel):
class KnowledgeGraphEdge(BaseModel):
id: str
type: str
type: Optional[str]
source: str # id of source node
target: str # id of target node
properties: dict[str, Any] # anything else goes here