mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-14 07:55:03 -05:00
09e01eddf4
### CHANGES - Introduce local `chat` package for message abstraction - Replace sashabaranov/go-openai with official openai-go SDK - Update OpenAI, Azure, and Exolab plugins for new client - Refactor all AI providers to use internal chat types - Decouple codebase from third-party AI provider structs - Replace deprecated `ioutil` functions with `os` equivalents
39 lines
1002 B
Go
39 lines
1002 B
Go
package fsdb
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/danielmiessler/fabric/chat"
|
|
)
|
|
|
|
func TestSessions_GetOrCreateSession(t *testing.T) {
|
|
dir := t.TempDir()
|
|
sessions := &SessionsEntity{
|
|
StorageEntity: &StorageEntity{Dir: dir, FileExtension: ".json"},
|
|
}
|
|
sessionName := "testSession"
|
|
session, err := sessions.Get(sessionName)
|
|
if err != nil {
|
|
t.Fatalf("failed to get or create session: %v", err)
|
|
}
|
|
if session.Name != sessionName {
|
|
t.Errorf("expected session name %v, got %v", sessionName, session.Name)
|
|
}
|
|
}
|
|
|
|
func TestSessions_SaveSession(t *testing.T) {
|
|
dir := t.TempDir()
|
|
sessions := &SessionsEntity{
|
|
StorageEntity: &StorageEntity{Dir: dir, FileExtension: ".json"},
|
|
}
|
|
sessionName := "testSession"
|
|
session := &Session{Name: sessionName, Messages: []*chat.ChatCompletionMessage{{Content: "message1"}}}
|
|
err := sessions.SaveSession(session)
|
|
if err != nil {
|
|
t.Fatalf("failed to save session: %v", err)
|
|
}
|
|
if !sessions.Exists(sessionName) {
|
|
t.Errorf("expected session to be saved")
|
|
}
|
|
}
|