jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: support LLM jinja2 template prompt (#3968)

Browse source 
Co-authored-by: Joel <iamjoel007@gmail.com>
This commit is contained in:
Yeuoly 2024-05-10 18:08:32 +08:00 • committed by GitHub
commit 8578ee0864
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
8 changed files with 325 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

4
api/core/prompt/entities/advanced_prompt_entities.py
View file

@ -1,4 +1,4 @@
from typing import Optional
from typing import Literal, Optional
from pydantic import BaseModel
@ -11,6 +11,7 @@ class ChatModelMessage(BaseModel):
"""
text: str
role: PromptMessageRole
edition_type: Optional[Literal['basic', 'jinja2']]
class CompletionModelPromptTemplate(BaseModel):
@ -18,6 +19,7 @@ class CompletionModelPromptTemplate(BaseModel):
Completion Model Prompt Template.
"""
text: str
edition_type: Optional[Literal['basic', 'jinja2']]
class MemoryConfig(BaseModel):