refactor: add ai agents and implement extract people agent

This commit is contained in:
2025-11-12 00:27:57 +08:00
parent 3d13aa18ae
commit 0a749d56e8
5 changed files with 617 additions and 0 deletions

22
src/agents/base_agent.py Normal file
View File

@@ -0,0 +1,22 @@
from langchain_openai import ChatOpenAI
from utils.config import get_instance as get_config
class BaseAgent:
def __init__(self, api_url: str = None, api_key: str = None, model_name: str = None):
config = get_config()
llm_api_url = api_url or config.get("ai", "llm_api_url")
llm_api_key = api_key or config.get("ai", "llm_api_key")
llm_model_name = model_name or config.get("ai", "llm_model_name")
self.llm = ChatOpenAI(
openai_api_key=llm_api_key,
openai_api_base=llm_api_url,
model_name=llm_model_name,
)
pass
class SummaryPeopleAgent(BaseAgent):
def __init__(self):
super().__init__()
pass