mirror of
https://github.com/langgenius/dify.git
synced 2026-03-15 06:09:43 +08:00
8a3485454a
- Move enqueue_draft_node_execution_trace import inside call site in workflow_service.py - Make gateway.py enterprise type imports lazy (routing dicts built on first call) - Restore typed ModelConfig in llm_generator method signatures (revert dict regression) - Fix generate_structured_output using wrong key model_parameters -> completion_params - Replace unsafe cast(str, msg.content) with get_text_content() across llm_generator - Remove duplicated payload classes from generator.py, import from core.llm_generator.entities - Gate _lookup_app_and_workspace_names and credential lookups in ops_trace_manager behind is_enterprise_telemetry_enabled()
23 lines
1.0 KiB
Python
23 lines
1.0 KiB
Python
"""Shared payload models for LLM generator helpers and controllers."""
|
|
|
|
from pydantic import BaseModel, Field
|
|
|
|
from core.app.app_config.entities import ModelConfig
|
|
|
|
|
|
class RuleGeneratePayload(BaseModel):
|
|
instruction: str = Field(..., description="Rule generation instruction")
|
|
model_config_data: ModelConfig = Field(..., alias="model_config", description="Model configuration")
|
|
no_variable: bool = Field(default=False, description="Whether to exclude variables")
|
|
app_id: str | None = Field(default=None, description="App ID for prompt generation tracing")
|
|
|
|
|
|
class RuleCodeGeneratePayload(RuleGeneratePayload):
|
|
code_language: str = Field(default="javascript", description="Programming language for code generation")
|
|
|
|
|
|
class RuleStructuredOutputPayload(BaseModel):
|
|
instruction: str = Field(..., description="Structured output generation instruction")
|
|
model_config_data: ModelConfig = Field(..., alias="model_config", description="Model configuration")
|
|
app_id: str | None = Field(default=None, description="App ID for prompt generation tracing")
|