Support for VertexAI custom models (Endpoints) - Githubissues

spring-projects / spring-ai

An Application Framework for AI Engineering

https://docs.spring.io/spring-ai/reference/1.0-SNAPSHOT/index.html

Apache License 2.0

2.84k stars 709 forks source link

Support for VertexAI custom models (Endpoints) #654

Open rafal-dudek opened 4 months ago

rafal-dudek commented 4 months ago

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

markpollack commented 1 month ago

There has been some work in this area to support models deployed on GKE. Will schedule for M2 in case it comes together in time.