OpenAI Plugin MVP (#38)

* OpenAI Plugin MVP Signed-off-by: Alex Meijer <[email protected]> * code review fixes Signed-off-by: Alex Meijer <[email protected]> * add MAINTAINERS Signed-off-by: Alex Meijer <[email protected]> * only attempt to read if no error Signed-off-by: Alex Meijer <[email protected]> --------- Signed-off-by: Alex Meijer <[email protected]>
opencost · Oct 10, 2024 · 0c42e1a · 0c42e1a
1 parent 42d14e3
commit 0c42e1a
Show file tree

Hide file tree

Showing 9 changed files with 906 additions and 0 deletions.
diff --git a/MAINTAINERS.md b/MAINTAINERS.md
@@ -0,0 +1,10 @@
+# OpenCost Plugins Committers and Maintainers
+
+Official list of OpenCost Plugins Maintainers
+
+## Maintainers
+
+| Maintainer | GitHub ID | Affiliation | Email |
+| --------------- | --------- | ----------- | ----------- |
+| Nik Willwerth | @nik-kc | Kubecost | <[email protected]> |
+| Alex Meijer | @ameijer | Kubecost | <[email protected]> |
diff --git a/pkg/plugins/openai/cmd/main/getcustomcosts_test.go b/pkg/plugins/openai/cmd/main/getcustomcosts_test.go
@@ -0,0 +1,53 @@
+package main
+
+import (
+ "os"
+ "testing"
+ "time"
+
+ openaiplugin "github.com/opencost/opencost-plugins/pkg/plugins/openai/openaiplugin"
+ "github.com/opencost/opencost/core/pkg/log"
+ "github.com/opencost/opencost/core/pkg/model/pb"
+ "github.com/opencost/opencost/core/pkg/util/timeutil"
+ "golang.org/x/time/rate"
+ "google.golang.org/protobuf/types/known/durationpb"
+ "google.golang.org/protobuf/types/known/timestamppb"
+)
+
+func TestGetCustomCosts(t *testing.T) {
+ // read necessary env vars. If any are missing, log warning and skip test
+ oaiApiKey := os.Getenv("OAI_API_KEY")
+ if oaiApiKey == "" {
+ log.Warnf("OAI_API_KEY undefined, skipping test")
+ t.Skip()
+ return
+ }
+
+ //set up config
+ config := openaiplugin.OpenAIConfig{
+ APIKey: oaiApiKey,
+ }
+
+ rateLimiter := rate.NewLimiter(1, 5)
+ oaiCostSrc := OpenAICostSource{
+ rateLimiter: rateLimiter,
+ config: &config,
+ }
+
+ windowStart := time.Date(2024, 10, 9, 0, 0, 0, 0, time.UTC)
+ // query for qty 2 of 1 hour windows
+ windowEnd := time.Date(2024, 10, 10, 0, 0, 0, 0, time.UTC)
+
+ req := &pb.CustomCostRequest{
+ Start: timestamppb.New(windowStart),
+ End: timestamppb.New(windowEnd),
+ Resolution: durationpb.New(timeutil.Day),
+ }
+
+ log.SetLogLevel("debug")
+ resp := oaiCostSrc.GetCustomCosts(req)
+
+ if len(resp) == 0 {
+ t.Fatalf("empty response")
+ }
+}