|
|
|
@@ -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 |
|
|
|
""" |