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
OpenDAS
ollama
Commits
51b2fd29
"docs/git@developer.sourcefind.cn:one/TransferBench.git" did not exist on "b56217d03f71463f9c4f262dfc63036c9425b278"
Unverified
Commit
51b2fd29
authored
Jul 19, 2024
by
royjhan
Committed by
GitHub
Jul 19, 2024
Browse files
adjust openai chat msg processing (#5729)
parent
d0634b15
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
9 additions
and
6 deletions
+9
-6
openai/openai.go
openai/openai.go
+3
-4
openai/openai_test.go
openai/openai_test.go
+6
-2
No files found.
openai/openai.go
View file @
51b2fd29
...
@@ -351,7 +351,6 @@ func fromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
...
@@ -351,7 +351,6 @@ func fromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
case
string
:
case
string
:
messages
=
append
(
messages
,
api
.
Message
{
Role
:
msg
.
Role
,
Content
:
content
})
messages
=
append
(
messages
,
api
.
Message
{
Role
:
msg
.
Role
,
Content
:
content
})
case
[]
any
:
case
[]
any
:
message
:=
api
.
Message
{
Role
:
msg
.
Role
}
for
_
,
c
:=
range
content
{
for
_
,
c
:=
range
content
{
data
,
ok
:=
c
.
(
map
[
string
]
any
)
data
,
ok
:=
c
.
(
map
[
string
]
any
)
if
!
ok
{
if
!
ok
{
...
@@ -363,7 +362,7 @@ func fromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
...
@@ -363,7 +362,7 @@ func fromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
if
!
ok
{
if
!
ok
{
return
nil
,
fmt
.
Errorf
(
"invalid message format"
)
return
nil
,
fmt
.
Errorf
(
"invalid message format"
)
}
}
message
.
Content
=
text
message
s
=
append
(
messages
,
api
.
Message
{
Role
:
msg
.
Role
,
Content
:
text
})
case
"image_url"
:
case
"image_url"
:
var
url
string
var
url
string
if
urlMap
,
ok
:=
data
[
"image_url"
]
.
(
map
[
string
]
any
);
ok
{
if
urlMap
,
ok
:=
data
[
"image_url"
]
.
(
map
[
string
]
any
);
ok
{
...
@@ -395,12 +394,12 @@ func fromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
...
@@ -395,12 +394,12 @@ func fromChatRequest(r ChatCompletionRequest) (*api.ChatRequest, error) {
if
err
!=
nil
{
if
err
!=
nil
{
return
nil
,
fmt
.
Errorf
(
"invalid message format"
)
return
nil
,
fmt
.
Errorf
(
"invalid message format"
)
}
}
message
.
Images
=
append
(
message
.
Images
,
img
)
messages
=
append
(
messages
,
api
.
Message
{
Role
:
msg
.
Role
,
Images
:
[]
api
.
ImageData
{
img
}})
default
:
default
:
return
nil
,
fmt
.
Errorf
(
"invalid message format"
)
return
nil
,
fmt
.
Errorf
(
"invalid message format"
)
}
}
}
}
messages
=
append
(
messages
,
message
)
default
:
default
:
if
msg
.
ToolCalls
==
nil
{
if
msg
.
ToolCalls
==
nil
{
return
nil
,
fmt
.
Errorf
(
"invalid message content type: %T"
,
content
)
return
nil
,
fmt
.
Errorf
(
"invalid message content type: %T"
,
content
)
...
...
openai/openai_test.go
View file @
51b2fd29
...
@@ -161,8 +161,12 @@ func TestMiddlewareRequests(t *testing.T) {
...
@@ -161,8 +161,12 @@ func TestMiddlewareRequests(t *testing.T) {
img
,
_
:=
base64
.
StdEncoding
.
DecodeString
(
imageURL
[
len
(
prefix
)
:
])
img
,
_
:=
base64
.
StdEncoding
.
DecodeString
(
imageURL
[
len
(
prefix
)
:
])
if
!
bytes
.
Equal
(
chatReq
.
Messages
[
0
]
.
Images
[
0
],
img
)
{
if
chatReq
.
Messages
[
1
]
.
Role
!=
"user"
{
t
.
Fatalf
(
"expected image encoding, got %s"
,
chatReq
.
Messages
[
0
]
.
Images
[
0
])
t
.
Fatalf
(
"expected 'user', got %s"
,
chatReq
.
Messages
[
1
]
.
Role
)
}
if
!
bytes
.
Equal
(
chatReq
.
Messages
[
1
]
.
Images
[
0
],
img
)
{
t
.
Fatalf
(
"expected image encoding, got %s"
,
chatReq
.
Messages
[
1
]
.
Images
[
0
])
}
}
},
},
},
},
...
...
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