jprdonnelly/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2025-12-19 17:27:16 -05:00
Code Issues Packages Projects Releases Wiki Activity

refactor: split changes for api/controllers/console/explore/completio… (#29894)

Browse Source
This commit is contained in:
Asuka Minato
2025-12-19 18:51:13 +09:00
committed by GitHub
parent 054d3f0da5
commit 4320503209
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
api/controllers/console/explore/completion.py
View File

@@ -40,7 +40,7 @@ from .. import console_ns
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
class CompletionMessagePayload(BaseModel): class CompletionMessageExplorePayload(BaseModel):
inputs: dict[str, Any] inputs: dict[str, Any]
query: str = "" query: str = ""
files: list[dict[str, Any]] | None = None files: list[dict[str, Any]] | None = None
@@ -71,7 +71,7 @@ class ChatMessagePayload(BaseModel):
raise ValueError("must be a valid UUID") from exc raise ValueError("must be a valid UUID") from exc
register_schema_models(console_ns, CompletionMessagePayload, ChatMessagePayload) register_schema_models(console_ns, CompletionMessageExplorePayload, ChatMessagePayload)
# define completion api for user # define completion api for user
@@ -80,13 +80,13 @@ register_schema_models(console_ns, CompletionMessagePayload, ChatMessagePayload)
endpoint="installed_app_completion", endpoint="installed_app_completion",
) )
class CompletionApi(InstalledAppResource): class CompletionApi(InstalledAppResource):
@console_ns.expect(console_ns.models[CompletionMessagePayload.__name__]) @console_ns.expect(console_ns.models[CompletionMessageExplorePayload.__name__])
def post(self, installed_app): def post(self, installed_app):
app_model = installed_app.app app_model = installed_app.app
if app_model.mode != AppMode.COMPLETION: if app_model.mode != AppMode.COMPLETION:
raise NotCompletionAppError() raise NotCompletionAppError()
payload = CompletionMessagePayload.model_validate(console_ns.payload or {}) payload = CompletionMessageExplorePayload.model_validate(console_ns.payload or {})
args = payload.model_dump(exclude_none=True) args = payload.model_dump(exclude_none=True)
streaming = payload.response_mode == "streaming" streaming = payload.response_mode == "streaming"