additional tools

This commit is contained in:
Kujtim Hoxha
2025-03-25 13:04:36 +01:00
parent 005b8ac167
commit 904061c243
33 changed files with 3258 additions and 236 deletions

View File

@@ -88,7 +88,6 @@ func (s *service) handleRequest(id string, sessionID string, content string) {
return
}
log.Printf("Request: %s", content)
currentAgent, systemMessage, err := agent.GetAgent(s.ctx, viper.GetString("agents.default"))
if err != nil {
s.Publish(AgentErrorEvent, AgentEvent{
@@ -172,7 +171,6 @@ func (s *service) handleRequest(id string, sessionID string, content string) {
}
session.PromptTokens += int64(usage.PromptTokens)
session.CompletionTokens += int64(usage.CompletionTokens)
// TODO: calculate cost
model := models.SupportedModels[models.ModelID(viper.GetString("models.big"))]
session.Cost += float64(usage.PromptTokens)*(model.CostPer1MIn/1_000_000) +
float64(usage.CompletionTokens)*(model.CostPer1MOut/1_000_000)