JohnCosta27/Haystack
1
0
Fork 0
Code Issues 7 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

fix(types): agent processing stuff

Browse Source
This commit is contained in:
John Costa
2025-04-09 12:12:09 +01:00
parent 5502fc6b19
commit 88fda32125
6 changed files with 24 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/agents/client/chat.go
View File

@ -73,7 +73,7 @@ type ChatAiMessage struct {
Role string `json:"role"`
ToolCalls *[]ToolCall `json:"tool_calls,omitempty"`
MessageContent `json:"content"`
Content string `json:"content"`
}
func (m ChatAiMessage) IsResponse() bool {

1
backend/agents/client/client.go
View File

@ -4,7 +4,6 @@ import (
"bytes"
"encoding/json"
"errors"
"fmt"
"io"
"log"
"net/http"

15
backend/agents/event_location_agent.go
View File

@ -161,16 +161,15 @@ func (agent EventLocationAgent) GetLocations(userId uuid.UUID, imageId uuid.UUID
ResponseFormat: client.ResponseFormat{
Type: "text",
},
Chat: client.Chat{
Messages: make([]client.ChatMessage, 0),
},
}
err = request.AddSystem(eventLocationPrompt)
if err != nil {
return err
}
request.Chat.AddSystem(eventLocationPrompt)
request.Chat.AddImage(imageName, imageData)
request.AddImage(imageName, imageData)
_, err = agent.client.Request(&request)
_, err = agent.client.Request(&request.Chat)
if err != nil {
return err
}
@ -180,7 +179,7 @@ func (agent EventLocationAgent) GetLocations(userId uuid.UUID, imageId uuid.UUID
UserId: userId,
}
return agent.client.Process(toolHandlerInfo, request)
return agent.client.Process(toolHandlerInfo, &request.Chat)
}
func NewLocationEventAgent(locationModel models.LocationModel, eventModel models.EventModel, contactModel models.ContactModel) (EventLocationAgent, error) {

12
backend/agents/note_agent.go
View File

@ -32,15 +32,15 @@ func (agent NoteAgent) GetNotes(userId uuid.UUID, imageId uuid.UUID, imageName s
ResponseFormat: client.ResponseFormat{
Type: "text",
},
Chat: client.Chat{
Messages: make([]client.ChatMessage, 0),
},
}
err := request.AddSystem(noteAgentPrompt)
if err != nil {
return err
}
request.Chat.AddSystem(noteAgentPrompt)
request.Chat.AddImage(imageName, imageData)
request.AddImage(imageName, imageData)
resp, err := agent.client.Request(&request)
resp, err := agent.client.Request(&request.Chat)
if err != nil {
return err
}

15
backend/agents/orchestrator.go
View File

@ -101,15 +101,16 @@ func (agent OrchestratorAgent) Orchestrate(userId uuid.UUID, imageId uuid.UUID,
},
ToolChoice: &toolChoice,
Tools: &tools,
Chat: client.Chat{
Messages: make([]client.ChatMessage, 0),
},
}
err = request.AddSystem(orchestratorPrompt)
if err != nil {
return err
}
request.Chat.AddSystem(orchestratorPrompt)
request.Chat.AddImage(imageName, imageData)
request.AddImage(imageName, imageData)
_, err = agent.client.Request(&request)
_, err = agent.client.Request(&request.Chat)
if err != nil {
return err
}
@ -119,7 +120,7 @@ func (agent OrchestratorAgent) Orchestrate(userId uuid.UUID, imageId uuid.UUID,
UserId: userId,
}
return agent.client.Process(toolHandlerInfo, request)
return agent.client.Process(toolHandlerInfo, &request.Chat)
}
func NewOrchestratorAgent(eventLocationAgent EventLocationAgent, noteAgent NoteAgent, imageName string, imageData []byte) (OrchestratorAgent, error) {

4
backend/main.go
View File

@ -25,10 +25,10 @@ import (
)
type TestAiClient struct {
ImageInfo client.ImageInfo
ImageInfo client.ImageMessageContent
}
func (client TestAiClient) GetImageInfo(imageName string, imageData []byte) (client.ImageInfo, error) {
func (client TestAiClient) GetImageInfo(imageName string, imageData []byte) (client.ImageMessageContent, error) {
return client.ImageInfo, nil
}