zlei6/ragflow
1
0
Fork 0
mirror of https://github.com/infiniflow/ragflow.git synced 2026-09-07 17:58:55 +08:00
Code Issues Packages Projects Releases Wiki Activity

Port agent PRs to GO - 5 (#16667)

Browse Source 
### Summary

Port 

https://github.com/infiniflow/ragflow/pull/15376
https://github.com/infiniflow/ragflow/pull/16401
https://github.com/infiniflow/ragflow/pull/15484
https://github.com/infiniflow/ragflow/pull/16685
This commit is contained in:
qinling0210
2026-07-08 19:54:29 +08:00
committed by GitHub
parent 0f08dc070d
commit 8e3bbad4da
5 changed files with 220 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
internal/service/openai_chat.go
View File

@@ -78,7 +78,7 @@ type OpenAICompletionResponse struct {
PromptTokens int
CompletionTokens int
TotalTokens int
Created int64
Created *int64
}
// OpenAIStreamEventKind discriminates stream events.
@@ -434,7 +434,6 @@ func (s *OpenAIChatService) OpenAIChatCompletions(c *gin.Context, userID, chatID
content := strings.TrimSpace(finalResult.Answer)
completionTokens := tokenizer.NumTokensFromString(content)
resp := &OpenAICompletionResponse{
Created: time.Now().Unix(),
Model: openaiReq.Model,
Content: content,
PromptTokens: promptTokens,
@@ -473,7 +472,7 @@ func (s *OpenAIChatService) OpenAIChatCompletions(c *gin.Context, userID, chatID
c.JSON(http.StatusOK, gin.H{
"id": completionID,
"object": "chat.completion",
"created": resp.Created,
"created": getCreatedOrDefault(resp.Created),
"model": resp.Model,
"usage": gin.H{
"prompt_tokens": resp.PromptTokens,
@@ -836,3 +835,10 @@ func (s *OpenAIChatService) writeArgError(c *gin.Context, msg string) {
func (s *OpenAIChatService) writeDataError(c *gin.Context, msg string) {
common.ResponseWithCodeData(c, common.CodeDataError, nil, msg)
}
func getCreatedOrDefault(created *int64) int64 {
if created != nil {
return *created
}
return time.Now().Unix()
}