Skip to content
Toggle navigation
P
Projects
G
Groups
S
Snippets
Help
phsl
/
new-api
This project
Loading...
Sign in
Toggle navigation
Go to a project
Project
Repository
Issues
0
Merge Requests
0
Pipelines
Wiki
Snippets
Members
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Commit
991b6f8b
authored
Feb 22, 2025
by
1808837298@qq.com
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
fix: mistral
parent
0333576b
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
13 additions
and
14 deletions
+13
-14
dto/openai_request.go
+1
-1
relay/channel/mistral/adaptor.go
+1
-1
relay/channel/mistral/text.go
+8
-12
relay/channel/openai/relay-openai.go
+3
-0
No files found.
dto/openai_request.go
View file @
991b6f8b
...
@@ -101,7 +101,7 @@ type Message struct {
...
@@ -101,7 +101,7 @@ type Message struct {
type
MediaContent
struct
{
type
MediaContent
struct
{
Type
string
`json:"type"`
Type
string
`json:"type"`
Text
string
`json:"text"`
Text
string
`json:"text
,omitempty
"`
ImageUrl
any
`json:"image_url,omitempty"`
ImageUrl
any
`json:"image_url,omitempty"`
InputAudio
any
`json:"input_audio,omitempty"`
InputAudio
any
`json:"input_audio,omitempty"`
}
}
...
...
relay/channel/mistral/adaptor.go
View file @
991b6f8b
...
@@ -41,7 +41,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, re
...
@@ -41,7 +41,7 @@ func (a *Adaptor) ConvertRequest(c *gin.Context, info *relaycommon.RelayInfo, re
if
request
==
nil
{
if
request
==
nil
{
return
nil
,
errors
.
New
(
"request is nil"
)
return
nil
,
errors
.
New
(
"request is nil"
)
}
}
return
request
,
nil
return
request
OpenAI2Mistral
(
request
)
,
nil
}
}
func
(
a
*
Adaptor
)
ConvertRerankRequest
(
c
*
gin
.
Context
,
relayMode
int
,
request
dto
.
RerankRequest
)
(
any
,
error
)
{
func
(
a
*
Adaptor
)
ConvertRerankRequest
(
c
*
gin
.
Context
,
relayMode
int
,
request
dto
.
RerankRequest
)
(
any
,
error
)
{
...
...
relay/channel/mistral/text.go
View file @
991b6f8b
package
mistral
package
mistral
import
(
import
(
"encoding/json"
"one-api/dto"
"one-api/dto"
)
)
func
requestOpenAI2Mistral
(
request
dto
.
GeneralOpenAIRequest
)
*
dto
.
GeneralOpenAIRequest
{
func
requestOpenAI2Mistral
(
request
*
dto
.
GeneralOpenAIRequest
)
*
dto
.
GeneralOpenAIRequest
{
messages
:=
make
([]
dto
.
Message
,
0
,
len
(
request
.
Messages
))
messages
:=
make
([]
dto
.
Message
,
0
,
len
(
request
.
Messages
))
for
_
,
message
:=
range
request
.
Messages
{
for
_
,
message
:=
range
request
.
Messages
{
if
!
message
.
IsStringContent
()
{
mediaMessages
:=
message
.
ParseContent
()
mediaMessages
:=
message
.
ParseContent
()
for
j
,
mediaMessage
:=
range
mediaMessages
{
for
j
,
mediaMessage
:=
range
mediaMessages
{
if
mediaMessage
.
Type
==
dto
.
ContentTypeImageURL
{
if
mediaMessage
.
Type
==
dto
.
ContentTypeImageURL
{
imageUrl
:=
mediaMessage
.
ImageUrl
.
(
dto
.
MessageImageUrl
)
imageUrl
:=
mediaMessage
.
ImageUrl
.
(
dto
.
MessageImageUrl
)
mediaMessage
.
ImageUrl
=
imageUrl
.
Url
mediaMessage
.
ImageUrl
=
imageUrl
.
Url
mediaMessages
[
j
]
=
mediaMessage
mediaMessages
[
j
]
=
mediaMessage
}
}
}
messageRaw
,
_
:=
json
.
Marshal
(
mediaMessages
)
message
.
Content
=
messageRaw
}
}
message
.
SetMediaContent
(
mediaMessages
)
messages
=
append
(
messages
,
dto
.
Message
{
messages
=
append
(
messages
,
dto
.
Message
{
Role
:
message
.
Role
,
Role
:
message
.
Role
,
Content
:
message
.
Content
,
Content
:
message
.
Content
,
...
...
relay/channel/openai/relay-openai.go
View file @
991b6f8b
...
@@ -87,6 +87,9 @@ func OaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
...
@@ -87,6 +87,9 @@ func OaiStreamHandler(c *gin.Context, resp *http.Response, info *relaycommon.Rel
info
.
SetFirstResponseTime
()
info
.
SetFirstResponseTime
()
ticker
.
Reset
(
time
.
Duration
(
constant
.
StreamingTimeout
)
*
time
.
Second
)
ticker
.
Reset
(
time
.
Duration
(
constant
.
StreamingTimeout
)
*
time
.
Second
)
data
:=
scanner
.
Text
()
data
:=
scanner
.
Text
()
if
common
.
DebugEnabled
{
println
(
data
)
}
if
len
(
data
)
<
6
{
// ignore blank line or wrong format
if
len
(
data
)
<
6
{
// ignore blank line or wrong format
continue
continue
}
}
...
...
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