spring-projects/spring-ai

Support for VertexAI custom models (Endpoints)

Open

#654 aperta il 29 apr 2024

Vedi su GitHub
 (2 commenti) (0 reazioni) (1 assegnatario)Java (2617 fork)auto 404
gcphelp wantedintegration testingvertex

Metriche repository

Star
 (8872 star)
Metriche merge PR
 (Merge medio 2g 1h) (50 PR mergiate in 30 g)

Descrizione

Expected Behavior

I would like to be able to use my Custom Model (e.g. fine-tuned foundation model) deployed on GCP Endpoint with Spring AI library.

API for Endpoints: https://cloud.google.com/vertex-ai/docs/reference/rest/v1/projects.locations.endpoints Example for predict: https://us-central1-aiplatform.googleapis.com/v1/projects/49...78/locations/us-central1/endpoints/433...736:predict {"instances":[{"content":"Hello"}],"parameters":{"temperature":0.2,"maxOutputTokens":1024,"topP":0.8,"topK":40,"candidateCount":1}}

Current Behavior

Currently in VertexAI PaLM2 Chat and VertexAI Gemini Chat you can only use Foundation models: https://cloud.google.com/vertex-ai/generative-ai/docs/model-reference/text-chat

Context

There is a similar issue for langchain4j: https://github.com/langchain4j/langchain4j/issues/440

Guida contributor