|
@@ -27,13 +27,13 @@ from core.orchestrator_rule_parser import OrchestratorRuleParser
|
|
|
from core.prompt.prompt_builder import PromptBuilder
|
|
|
from core.prompt.prompt_template import JinjaPromptTemplate
|
|
|
from core.prompt.prompts import MORE_LIKE_THIS_GENERATE_PROMPT
|
|
|
-from models.model import App, AppModelConfig, Account, Conversation, Message
|
|
|
+from models.model import App, AppModelConfig, Account, Conversation, Message, EndUser
|
|
|
|
|
|
|
|
|
class Completion:
|
|
|
@classmethod
|
|
|
def generate(cls, task_id: str, app: App, app_model_config: AppModelConfig, query: str, inputs: dict,
|
|
|
- user: Account, conversation: Optional[Conversation], streaming: bool, is_override: bool = False):
|
|
|
+ user: Union[Account, EndUser], conversation: Optional[Conversation], streaming: bool, is_override: bool = False):
|
|
|
"""
|
|
|
errors: ProviderTokenNotInitError
|
|
|
"""
|