mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-04 12:56:02 +00:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""Dify Knowledge Query Node - query Dify knowledge base
|
|
|
|
Node metadata is loaded from: ../../templates/metadata/nodes/dify_knowledge_query.yaml
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
from langbot_plugin.api.entities.builtin.workflow import ExecutionContext
|
|
from ..node import WorkflowNode, workflow_node
|
|
|
|
@workflow_node('dify_knowledge_query')
|
|
class DifyKnowledgeQueryNode(WorkflowNode):
|
|
"""Dify knowledge base query node - query Dify knowledge base"""
|
|
|
|
category = 'integration'
|
|
|
|
async def execute(self, inputs: dict[str, Any], context: ExecutionContext) -> dict[str, Any]:
|
|
base_url = self.get_config('base_url', 'https://api.dify.ai/v1')
|
|
api_key = self.get_config('api_key', '')
|
|
dataset_id = self.get_config('dataset_id', '')
|
|
query = inputs.get('query', '')
|
|
|
|
return {
|
|
'results': [],
|
|
'success': False,
|
|
'_debug': {
|
|
'base_url': base_url,
|
|
'api_key': api_key[:8] + '...' if api_key else '',
|
|
'dataset_id': dataset_id,
|
|
'query': query,
|
|
},
|
|
}
|