Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
open-webui
Commits
b9b1fdd1
Commit
b9b1fdd1
authored
Jul 31, 2024
by
Michael Poluektov
Browse files
refac: rename message_template
parent
2e0fa1c6
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
backend/utils/misc.py
backend/utils/misc.py
+3
-3
No files found.
backend/utils/misc.py
View file @
b9b1fdd1
...
@@ -88,7 +88,7 @@ def add_or_update_system_message(content: str, messages: List[dict]):
...
@@ -88,7 +88,7 @@ def add_or_update_system_message(content: str, messages: List[dict]):
return
messages
return
messages
def
message_template
(
model
:
str
):
def
openai_chat_
message_template
(
model
:
str
):
return
{
return
{
"id"
:
f
"
{
model
}
-
{
str
(
uuid
.
uuid4
())
}
"
,
"id"
:
f
"
{
model
}
-
{
str
(
uuid
.
uuid4
())
}
"
,
"created"
:
int
(
time
.
time
()),
"created"
:
int
(
time
.
time
()),
...
@@ -98,14 +98,14 @@ def message_template(model: str):
...
@@ -98,14 +98,14 @@ def message_template(model: str):
def
openai_chat_chunk_message_template
(
model
:
str
,
message
:
str
):
def
openai_chat_chunk_message_template
(
model
:
str
,
message
:
str
):
template
=
message_template
(
model
)
template
=
openai_chat_
message_template
(
model
)
template
[
"object"
]
=
"chat.completion.chunk"
template
[
"object"
]
=
"chat.completion.chunk"
template
[
"choices"
][
0
][
"delta"
]
=
{
"content"
:
message
}
template
[
"choices"
][
0
][
"delta"
]
=
{
"content"
:
message
}
return
template
return
template
def
openai_chat_completion_message_template
(
model
:
str
,
message
:
str
):
def
openai_chat_completion_message_template
(
model
:
str
,
message
:
str
):
template
=
message_template
(
model
)
template
=
openai_chat_
message_template
(
model
)
template
[
"object"
]
=
"chat.completion"
template
[
"object"
]
=
"chat.completion"
template
[
"choices"
][
0
][
"message"
]
=
{
"content"
:
message
,
"role"
:
"assistant"
}
template
[
"choices"
][
0
][
"message"
]
=
{
"content"
:
message
,
"role"
:
"assistant"
}
template
[
"choices"
][
0
][
"finish_reason"
]
=
"stop"
template
[
"choices"
][
0
][
"finish_reason"
]
=
"stop"
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment