spring-projects/spring-ai

Support for VertexAI custom models (Endpoints)

Open

#654 创建于 2024年4月29日

在 GitHub 查看
 (2 评论) (0 反应) (1 负责人)Java (2,617 fork)auto 404
gcphelp wantedintegration testingvertex

仓库指标

Star
 (8,872 star)
PR 合并指标
 (平均合并 2天 1小时) (30 天内合并 50 个 PR)

描述

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

贡献者指南