Merge pull request #1 from devalexandre/feat/add-simple-run

feat: add simple run

Author: devalexandre

client_test.go

@@ -4,7 +4,9 @@ import (
 	"context"
 	"fmt"
 	"log"
+	"os"
 	"testing"
+	"time"
 
 	"github.com/google/uuid"
 	"github.com/tmc/langchaingo/llms"
@@ -122,6 +124,7 @@ func TestRun(t *testing.T) {
 
 	// use 2 chains
 	t.Run("use with 2 traces", func(t *testing.T) {
+		t.Skip()
 		// Create a new client
 		runId := uuid.New().String()
 		client := NewClient("")
@@ -201,4 +204,92 @@ func TestRun(t *testing.T) {
 
 	})
 
+	t.Run("use with 2 traces and SimpleRun", func(t *testing.T) {
+		// Create a new client
+		runId := uuid.New().String()
+		client := NewClient(os.Getenv("LANGSMITH_API_KEY"))
+
+		opts := []openai.Option{
+			openai.WithModel("gpt-3.5-turbo-0125"),
+			openai.WithEmbeddingModel("text-embedding-3-large"),
+		}
+		llm, err := openai.New(opts...)
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		ctx := context.Background()
+
+		content := []llms.MessageContent{
+			llms.TextParts(llms.ChatMessageTypeSystem, "You are a company branding design wizard."),
+			llms.TextParts(llms.ChatMessageTypeHuman, "What would be a good company name a company that makes colorful socks?"),
+		}
+
+		if err != nil {
+			t.Errorf("Error running: %v", err)
+		}
+		startTime := time.Now().UTC()
+		out, err := llm.GenerateContent(ctx, content)
+		if err != nil {
+			t.Errorf("Error running: %v", err)
+
+		}
+		runPayload := &RunPayload{
+			Name:        "langsmithgo-llm",
+			SessionName: "langsmithgo",
+			RunType:     LLM,
+			RunID:       runId,
+			Tags:        []string{"llm"},
+			StartTime:   startTime,
+			Inputs: map[string]interface{}{
+				"prompt":      content, // Ensure 'output' is properly defined and is of type that has a String method
+				"model":       "gpt-3.5-turbo",
+				"temperature": 0.7, // Assuming 'temperature' should be a float, not a string
+			},
+			Outputs: map[string]interface{}{
+				"output": out,
+			},
+			EndTime: time.Now().UTC(),
+		}
+
+		err = client.RunSingle(runPayload)
+
+		if err != nil {
+			t.Errorf("Error running: %v", err)
+		}
+
+		embdId := uuid.New().String()
+		// create embedding
+
+		startTime = time.Now().UTC()
+
+		embedings, err := llm.CreateEmbedding(ctx, []string{"ola", "mundo"})
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		err = client.RunSingle(&RunPayload{
+			Name:        "langsmithgo-llm",
+			SessionName: "langsmithgo",
+			RunType:     Embedding,
+			StartTime:   startTime,
+			RunID:       embdId,
+			ParentID:    runId,
+			Tags:        []string{"llm"},
+			Inputs: map[string]interface{}{
+				"prompt":      out.Choices[0].Content, // Ensure 'output' is properly defined and is of type that has a String method
+				"model":       "gpt-3.5-turbo",
+				"temperature": 0.7, // Assuming 'temperature' should be a float, not a string
+			},
+			Outputs: map[string]interface{}{
+				"output": embedings,
+			},
+			EndTime: time.Now().UTC(),
+		})
+
+		if err != nil {
+			log.Fatal(err)
+		}
+
+	})
 }

cliente.go

@@ -34,7 +34,6 @@ func (c *Client) PostRun(input *RunPayload) error {
 		Tags:        input.Tags,
 		ParentId:    input.ParentID,
 		Extras:      input.Extras,
-		Metadata:    input.Metadata,
 	}
 
 	jsonData, err := json.Marshal(payload)
@@ -77,6 +76,41 @@ func (c *Client) Run(input *RunPayload) error {
 
 }
 
+/*
+Single Requests
+If you want to reduce the number of requests you make to LangSmith, you can log runs in a single request. Just be sure to include the outputs or error and fix the end_time all in the post request.
+Below is an example that logs the completion LLM run from above in a single call.
+*/
+
+func (c *Client) RunSingle(input *RunPayload) error {
+	payload := SimplePayload{
+		PostPayload: PostPayload{ // Especificando a struct PostPayload
+			ID:          input.RunID,
+			Name:        input.Name,
+			RunType:     input.RunType,
+			StartTime:   input.StartTime,
+			Inputs:      input.Inputs,
+			SessionName: input.SessionName,
+			Tags:        input.Tags,
+			ParentId:    input.ParentID,
+			Extras:      input.Extras,
+		},
+		PatchPayload: PatchPayload{ // Especificando a struct PatchPayload
+			Outputs: input.Outputs,
+			EndTime: input.EndTime,
+			Events:  input.Events,
+		},
+	}
+
+	jsonData, err := json.Marshal(payload)
+	if err != nil {
+		return err
+	}
+	err = c.Do(BASE_URL, http.MethodPost, jsonData)
+
+	return err
+}
+
 // client for http requests
 func (c *Client) Do(url string, method string, jsonData []byte) error {
 	req, err := http.NewRequest(method, url, bytes.NewBuffer(jsonData))

contracts.go

@@ -30,14 +30,18 @@ type RunPayload struct {
 	Outputs     map[string]interface{} `json:"outputs"`
 	EndTime     time.Time              `json:"end_time"`
 	Extras      map[string]interface{} `json:"extras"`
-	Metadata    map[string]interface{} `json:"metadata"`
 	Events      []Event                `json:"events"`
 }
 
 type Client struct {
 	APIKey string
 }
 
+type SimplePayload struct {
+	PostPayload
+	PatchPayload
+}
+
 type PostPayload struct {
 	ID          string                 `json:"id"`
 	Name        string                 `json:"name"`
@@ -47,8 +51,7 @@ type PostPayload struct {
 	SessionName string                 `json:"session_name"`
 	Tags        []string               `json:"tags,omitempty"`
 	ParentId    string                 `json:"parent_run_id,omitempty"`
-	Extras      map[string]interface{} `json:"extras,omitempty",`
-	Metadata    map[string]interface{} `json:"metadata,omitempty"`
+	Extras      map[string]interface{} `json:"extras,omitempty"`
 }
 
 type PatchPayload struct {