feat: backend, trade state

backend/handler/state.go

@@ -1,9 +1,17 @@
 package handler
 
-import "github.com/gofiber/fiber/v2"
+import (
+	"github.com/gofiber/fiber/v2"
+
+	"github.com/crestalnetwork/ethglobal-bangkok/backend/types"
+)
 
 func (h *Handler) GetChatState(c *fiber.Ctx) error {
-	resp, err := h.s.GetChatState(c.Context())
+	id := c.Params("id")
+	if id == "" {
+		return types.NewError(fiber.StatusBadRequest, "BadRequest", "missing id")
+	}
+	resp, err := h.s.GetChatState(c.Context(), id)
 	if err != nil {
 		return err
 	}

backend/service/state.go

@@ -11,7 +11,7 @@ func (s *Service) GetChatState(ctx context.Context, id string) (*types.State, er
 	raw, ok := s.state.Load(id)
 	if !ok {
 		resp = &types.State{
-			ChatID: id,
+			CallID: id,
 		}
 		s.state.Store(id, resp)
 		return resp, nil

backend/service/vapi_function.go

@@ -38,11 +38,24 @@ func (s *Service) VAPIFunctionTrade(ctx context.Context, msg *types.VapiServerMe
 			}
 		}
 	}
+
+	callID := msg.Message.Call.Id
+	// check the call id
+	state, err := s.GetChatState(ctx, callID)
+	if err != nil {
+		return nil, err
+	}
+	if state.Step > 0 {
+		return vapiToolResponse(id, fmt.Sprintf("You are trading %s, please confirm or cancel the transaction.", state.Trade.Currency)), nil
+	}
+
 	s.log.Warn("VAPIFunction called", "trade", trade)
-	res := types.ToolResult{ToolCallID: id, Result: fmt.Sprintf("Tell the user:The prize of %s is $%f now, are you sure you want to go ahead with this transaction?",
-		trade.Currency, rand.Float64()*100)}
-	resp.Results = append(resp.Results, res)
-	return resp, nil
+	trade.Price = rand.Float64() * 100 // mock the price
+	// update the state
+	state.Trade = *trade
+	state.Step = 1
+	s.state.Store(callID, state)
+	return vapiToolResponse(id, fmt.Sprintf("Tell the user: The prize of %s is $%f now, are you sure you want to go ahead with this transaction?", trade.Currency, trade.Price)), nil
 }
 
 func (s *Service) VAPIFunctionConfirm(ctx context.Context, msg *types.VapiServerMessageToolCall) (*types.ToolResults, error) {
@@ -117,3 +130,14 @@ func (s *Service) VAPIFunction(ctx context.Context, genericMessage map[string]in
 	// return &types.FunctionResult{Result: "Your transaction is in progress, please wait."}, nil
 	return nil
 }
+
+func vapiToolResponse(toolID string, result string) *types.ToolResults {
+	return &types.ToolResults{
+		Results: []types.ToolResult{
+			{
+				ToolCallID: toolID,
+				Result:     result,
+			},
+		},
+	}
+}

backend/types/state.go

@@ -1,7 +1,7 @@
 package types
 
 type State struct {
-	ChatID      string `json:"chat_id"`
+	CallID      string `json:"call_id"`
 	Step        int    `json:"step"`         // 0 start, 1 confirming, 2 signing, 3 trading, 4 done
 	Trade       Trade  `json:"trade"`        // start from 1
 	IsConfirmed bool   `json:"is_confirmed"` // start from 2

backend/types/vapi.go

@@ -66,6 +66,7 @@ type VapiServerMessageToolCall struct {
 }
 
 type VapiServerMessageToolCallMessage struct {
+	Call            vapi.Call        `json:"call"`
 	ToolCallList    []*vapi.ToolCall `json:"toolCallList,omitempty" url:"toolCallList,omitempty"`
 	extraProperties map[string]interface{}
 	_rawJSON        json.RawMessage