pangukitsappdev.retriever package¶

Submodules¶

pangukitsappdev.retriever.css_tool_retriever module¶

class pangukitsappdev.retriever.css_tool_retriever.CSSToolRetriever(tool_provider: ToolProvider, vector_store_config: VectorStoreConfig)¶

基类：AbstractToolRetriever

add_tools(tools: List[AbstractTool])¶: 工具入库 :param tools: tools工具

dsl_search(query: str, dsl: str) → List[AbstractTool]¶: 工具检索 :param query : 查询语句 :param dsl : dsl :return: 相似工具列表

preprocessor: Callable[[List[ConversationMessage]], str]¶

tool_provider: ToolProvider¶

vector: Optional[VectorApi]¶

vector_store_config: VectorStoreConfig¶

pangukitsappdev.retriever.kg_retriever module¶

class pangukitsappdev.retriever.kg_retriever.KGRetriever(*, name: Optional[str] = None, tags: Optional[List[str]] = None, metadata: Optional[Dict[str, Any]] = None, kg: KnowledgeGraph)¶

基类：BaseRetriever

kg: KnowledgeGraph¶

pangukitsappdev.retriever.retrieved_tool module¶

class pangukitsappdev.retriever.retrieved_tool.RetrievedTool(*, tool_id: str, tool_metadata: Dict[str, Any] = {}, score: float)¶

基类：ToolMetadata

工具的元数据信息 Attributes:

score: 评分，如从ES库检索后的score字段

score: float¶

pangukitsappdev.retriever.tool_metadata module¶

class pangukitsappdev.retriever.tool_metadata.ToolMetadata(*, tool_id: str, tool_metadata: Dict[str, Any] = {})¶

基类：BaseModel

工具的元数据信息 Attributes:

tool_id: 工具ID tool_metadata: 工具元数据，如从ES库中检索到的source字段

tool_id: str¶

tool_metadata: Dict[str, Any]¶

pangukitsappdev.retriever.tool_retriever_impl module¶

class pangukitsappdev.retriever.tool_retriever_impl.ToolRetrieverImpl(tool_provider: ToolProvider, vector_store_config: VectorStoreConfig)¶