relay_request.go 517 B

123456789101112131415161718192021222324252627
  1. package types
  2. type RelayRequest struct {
  3. OriginRequest any
  4. Format RelayFormat
  5. PromptTokenCount int
  6. }
  7. func (r *RelayRequest) CopyOriginRequest() any {
  8. if r.OriginRequest == nil {
  9. return nil
  10. }
  11. switch v := r.OriginRequest.(type) {
  12. case *GeneralOpenAIRequest:
  13. return v.Copy()
  14. case *GeneralClaudeRequest:
  15. return v.Copy()
  16. case *GeneralGeminiRequest:
  17. return v.Copy()
  18. case *GeneralRerankRequest:
  19. return v.Copy()
  20. case *GeneralEmbeddingRequest:
  21. return v.Copy()
  22. default:
  23. return nil
  24. }
  25. }