youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-04-22 11:47:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor app

Browse Source
This commit is contained in:
takatost
2024-02-29 17:33:52 +08:00
parent 896c200211
commit 799db69e4f
94 changed files with 991 additions and 721 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/core/callback_handler/agent_loop_gather_callback_handler.py
View File

@ -7,7 +7,7 @@ from langchain.agents import openai_functions_agent, openai_functions_multi_agen
from langchain.callbacks.base import BaseCallbackHandler
from langchain.schema import AgentAction, AgentFinish, BaseMessage, LLMResult
from core.application_queue_manager import ApplicationQueueManager, PublishFrom
from core.app.app_queue_manager import AppQueueManager, PublishFrom
from core.callback_handler.entity.agent_loop import AgentLoop
from core.entities.application_entities import ModelConfigEntity
from core.model_runtime.entities.llm_entities import LLMResult as RuntimeLLMResult
@ -22,7 +22,7 @@ class AgentLoopGatherCallbackHandler(BaseCallbackHandler):
raise_error: bool = True
def __init__(self, model_config: ModelConfigEntity,
queue_manager: ApplicationQueueManager,
queue_manager: AppQueueManager,
message: Message,
message_chain: MessageChain) -> None:
"""Initialize callback handler."""

4
api/core/callback_handler/index_tool_callback_handler.py
View File

@ -1,5 +1,5 @@
from core.application_queue_manager import ApplicationQueueManager, PublishFrom
from core.app.app_queue_manager import AppQueueManager, PublishFrom
from core.entities.application_entities import InvokeFrom
from core.rag.models.document import Document
from extensions.ext_database import db
@ -10,7 +10,7 @@ from models.model import DatasetRetrieverResource
class DatasetIndexToolCallbackHandler:
"""Callback handler for dataset tool."""
def __init__(self, queue_manager: ApplicationQueueManager,
def __init__(self, queue_manager: AppQueueManager,
app_id: str,
message_id: str,
user_id: str,