Function TestAdapterConformance_VLLM

internal/adapters/conformance_test.go:31–59

(t *testing.T)

}

func TestAdapterConformance_VLLM(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		switch r.URL.Path {
		case "/health":
			w.WriteHeader(http.StatusOK)
			_, _ = w.Write([]byte(`{"status":"ok"}`))
		case "/v1/chat/completions":
			w.Header().Set("Content-Type", "application/json")
			_ = json.NewEncoder(w).Encode(map[string]any{
				"choices": []map[string]any{
					{"message": map[string]any{"content": "hello from vllm"}},
				},
			})
		default:
			http.NotFound(w, r)
		}
	}))
	defer srv.Close()

	adapter := vllm.New(config.BackendConfig{
		Name:         "vllm-b",
		Type:         "vllm",
		Endpoint:     srv.URL,
		TimeoutMS:    200,
		Capabilities: []string{"chat"},
		Cost:         config.CostConfig{Unit: 2},
	})
	assertAdapterConformance(t, adapter)
}

func TestAdapterConformance_Anthropic(t *testing.T) {
	srv := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {