Skip to content

fix(go): Fixed Go schema generator. #2296

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
apascal07 wants to merge 1 commit into from
Closed

fix(go): Fixed Go schema generator. #2296

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion genkit-tools/common/src/types/model.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -257,7 +257,7 @@ export const GenerationUsageSchema = z.object({
export type GenerationUsage = z.infer<typeof GenerationUsageSchema>;

/** Model response finish reason enum. */
const FinishReasonSchema = z.enum([
export const FinishReasonSchema = z.enum([
'stop',
'length',
'blocked',
Expand Down
32 changes: 26 additions & 6 deletions go/core/schemas.config
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,7 @@ by the model in a [ToolRequest].
.

Candidate omit
CandidateFinishReason omit
CandidateFinishReason name FinishReason
DocumentData pkg ai
GenerateResponse omit
GenerateResponseChunk omit
Expand Down Expand Up @@ -161,26 +161,44 @@ TextPart.data omit
TextPart.media omit
TextPart.toolRequest omit
TextPart.toolResponse omit
TextPart.custom omit
TextPart.metadata type map[string]any
MediaPart pkg ai
MediaPart name mediaPart
MediaPart.text omit
MediaPart.toolRequest omit
MediaPart.toolResponse omit
MediaPart.custom omit
MediaPart.data omit
MediaPart.metadata type map[string]any
MediaPartMedia pkg ai
MediaPartMedia name mediaPartMedia
ToolRequestPart.text omit
ToolRequestPart.media omit
ToolRequestPart.data omit
ToolRequestPart.toolResponse omit
ToolRequestPart.custom omit
ToolRequestPart.metadata type map[string]any
ToolResponsePart.text omit
ToolResponsePart.media omit
ToolResponsePart.data omit
ToolResponsePart.toolRequest omit
ToolResponsePart.custom omit
ToolResponsePart.metadata type map[string]any
DataPart pkg ai
DataPart name dataPart
DataPart.media omit
DataPart.text omit
DataPart.toolRequest omit
DataPart.toolResponse omit
DataPart.custom omit
DataPart.metadata type map[string]any
CustomPart pkg ai
CustomPart name customPart
ModelInfo pkg ai
ModelInfoSupports pkg ai
ModelInfoSupports.output type OutputFormat
ModelInfoSupports.constrained omit
Role pkg ai
RoleUser pkg ai
RoleModel pkg ai
Expand All @@ -194,22 +212,23 @@ GenerateActionOptions.messages type []*Message
GenerateActionOptions.tools type []*ToolDefinition
GenerateActionOptionsToolChoice name ToolChoice
GenerateActionOptions.config type any
GenerateActionOptions.output type *GenerateActionOptionsOutput
GenerateActionOptions.output type *GenerateActionOutputConfig
GenerateActionOptions.returnToolRequests type bool
GenerateActionOptions.maxTurns type int

GenerateActionOptionsOutput.instructions omit
GenerateActionOptionsOutput.format type OutputFormat
GenerateActionOptionsOutput.jsonSchema type map[string]any
GenerateActionOutputConfig.instructions omit
GenerateActionOutputConfig.format type OutputFormat
GenerateActionOutputConfig.jsonSchema name Schema
GenerateActionOutputConfig.jsonSchema type map[string]any

# ModelRequest
ModelRequest pkg ai
ModelRequest.config type any
ModelRequest.context type []*Document
ModelRequest.messages type []*Message
ModelRequest.output type *ModelRequestOutput
ModelRequest.tools type []*ToolDefinition
ModelRequest.toolChoice type ToolChoice
ModelRequest.docs type []*Document

# ModelResponse
ModelResponse pkg ai
Expand All @@ -221,6 +240,7 @@ ModelResponse.latencyMs type float64
ModelResponse.message type *Message
ModelResponse.request type *ModelRequest
ModelResponse.usage type *GenerationUsage
ModelResponse.raw omit

# ModelResponseChunk
ModelResponseChunk pkg ai
Expand Down
2 changes: 1 addition & 1 deletion js/ai/src/model.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -279,7 +279,7 @@ export const GenerationUsageSchema = z.object({
export type GenerationUsage = z.infer<typeof GenerationUsageSchema>;

/** Model response finish reason enum. */
const FinishReasonSchema = z.enum([
export const FinishReasonSchema = z.enum([
'stop',
'length',
'blocked',
Expand Down
Loading