zishuo/apinto
1
0
Fork 0
mirror of https://github.com/eolinker/apinto synced 2025-12-24 13:28:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #180 from eolinker/feature/openai-demo

Browse Source 
add: validNormalFunc
This commit is contained in:
Dot.L
2024-12-26 10:52:38 +08:00
committed by GitHub
parent fb9c93c724 087fd45a1f
commit bb4dff6024
1 changed files with 25 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
drivers/ai-provider/openAI/openai_test.go
View File

@@ -7,6 +7,8 @@ import (
"testing"
"time"
"github.com/eolinker/eosc/eocontext"
"github.com/eolinker/apinto/convert"
ai_provider "github.com/eolinker/apinto/drivers/ai-provider"
http_context "github.com/eolinker/apinto/node/http-context"
@@ -41,6 +43,19 @@ var (
}`)
)
func validNormalFunc(ctx eocontext.EoContext) bool {
fmt.Printf("input token: %d\n", ai_provider.GetAIModelInputToken(ctx))
fmt.Printf("output token: %d\n", ai_provider.GetAIModelOutputToken(ctx))
fmt.Printf("total token: %d\n", ai_provider.GetAIModelTotalToken(ctx))
if ai_provider.GetAIModelInputToken(ctx) <= 0 {
return false
}
if ai_provider.GetAIModelOutputToken(ctx) <= 0 {
return false
}
return ai_provider.GetAIModelTotalToken(ctx) > 0
}
// TestSentTo tests the end-to-end execution of the OpenAI integration.
func TestSentTo(t *testing.T) {
// Load .env file
@@ -55,12 +70,14 @@ func TestSentTo(t *testing.T) {
apiKey string
wantStatus string
body []byte
validFunc func(ctx eocontext.EoContext) bool
}{
{
name: "success",
apiKey: os.Getenv("ValidKey"),
wantStatus: ai_provider.StatusNormal,
body: successBody,
validFunc: validNormalFunc,
},
{
name: "invalid request",
@@ -83,7 +100,7 @@ func TestSentTo(t *testing.T) {
// Run tests for each scenario
for _, data := range testData {
t.Run(data.name, func(t *testing.T) {
if err := runTest(data.apiKey, data.body, data.wantStatus); err != nil {
if err := runTest(data.apiKey, data.body, data.wantStatus, data.validFunc); err != nil {
t.Fatalf("Test failed: %v", err)
}
})
@@ -91,7 +108,7 @@ func TestSentTo(t *testing.T) {
}
// runTest handles a single test case
func runTest(apiKey string, requestBody []byte, wantStatus string) error {
func runTest(apiKey string, requestBody []byte, wantStatus string, validFunc func(ctx eocontext.EoContext) bool) error {
cfg := &Config{
APIKey: apiKey,
Base: "https://api.openai.com",
@@ -127,6 +144,12 @@ func runTest(apiKey string, requestBody []byte, wantStatus string) error {
if ai_provider.GetAIStatus(ctx) != wantStatus {
return fmt.Errorf("unexpected status: got %s, expected %s", ai_provider.GetAIStatus(ctx), wantStatus)
}
if validFunc != nil {
if validFunc(ctx) {
return nil
}
return fmt.Errorf("execute validFunc failed")
}
return nil
}