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

fix: adding extension hack

Browse Source 
fix
This commit is contained in:
John Costa
2025-05-10 22:01:17 +01:00
parent b97eae10a3
commit 5a2b990c0c
2 changed files with 10 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -216,6 +216,12 @@ func (chat *Chat) AddImage(imageName string, image []byte, query *string) error
index += 1 index += 1
} }
if len(extension) == 0 {
// Hacky! It seems apple doesnt add extension.
// BIG TODO: take better metadata from the image.
extension = "png"
}
messageContent.Content[index] = ImageMessageContent{ messageContent.Content[index] = ImageMessageContent{
ImageType: "image_url", ImageType: "image_url",
ImageUrl: ImageMessageUrl{ ImageUrl: ImageMessageUrl{

4
backend/main.go
View File

@ -91,6 +91,10 @@ func main() {
// TODO: this could be part of the db table // TODO: this could be part of the db table
extension := filepath.Ext(image.ImageName) extension := filepath.Ext(image.ImageName)
if len(extension) == 0 {
// Same hack
extension = "png"
}
extension = extension[1:] extension = extension[1:]
w.Header().Add("Content-Type", "image/"+extension) w.Header().Add("Content-Type", "image/"+extension)