fix: protect concurrent map access with mutexLock (#15)

Author: forrestIsRunning

traceloop-sdk/prompt_registry.go

@@ -31,9 +31,11 @@ func (instance *Traceloop) populatePromptRegistry() {
 		return
 	}
 
+	instance.registryMutex.Lock()
 	for _, prompt := range response.Prompts {
 		instance.promptRegistry[prompt.Key] = &prompt
 	}
+	instance.registryMutex.Unlock()
 }
 
 func (instance *Traceloop) pollPrompts() {
@@ -55,6 +57,8 @@ func (instance *Traceloop) pollPrompts() {
 }
 
 func (instance *Traceloop) getPromptVersion(key string) (*model.PromptVersion, error) {
+	instance.registryMutex.RLock()
+	defer instance.registryMutex.RUnlock()
 	if instance.promptRegistry[key] == nil {
 		return nil, fmt.Errorf("prompt with key %s not found", key)
 	}

traceloop-sdk/sdk.go

@@ -6,6 +6,7 @@ import (
 	"log"
 	"net/http"
 	"os"
+	"sync"
 	"time"
 
 	"go.opentelemetry.io/otel/attribute"
@@ -21,6 +22,7 @@ const PromptsPath = "/v1/traceloop/prompts"
 type Traceloop struct {
 	config         Config
 	promptRegistry model.PromptRegistry
+	registryMutex  sync.RWMutex
 	tracerProvider *trace.TracerProvider
 	http.Client
 }