mirror of
https://github.com/cheikh2shift/godex.git
synced 2026-10-19 01:29:53 -05:00
increase prompt token count
This commit is contained in:
1 parent
dc43d6dc51
commit
40be168ffb
1 file changed
+1
-1
@@ -880,7 +880,7 @@ func (p *llamaProvider) Send(ctx context.Context, prompt string) (string, error)
|
||||
"role": "assistant",
|
||||
"content": response,
|
||||
})
|
||||
p.promptTokens = totalPromptTokens
|
||||
p.promptTokens += totalPromptTokens
|
||||
p.completionTokens = totalCompletionTokens
|
||||
p.mu.Unlock()
|
||||
|
||||
|
||||
Reference in new issue
Block a user