Skip to content

Commit af606a1

Browse files
committed
Get rid of unnecessary GenerationConfig classes in favor of storing options directly on AiModelConfig.
1 parent 0f6e21f commit af606a1

File tree

1 file changed

+20
-26
lines changed

1 file changed

+20
-26
lines changed

docs/ARCHITECTURE.md

Lines changed: 20 additions & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -755,31 +755,31 @@ direction LR
755755
getRequiredOptions() AiRequiredOption[]
756756
}
757757
class AiModelConfig {
758+
+setOutputModalities(AiModality[] $modalities) void
759+
+getOutputModalities() AiModality[]
758760
+setSystemInstruction(string|MessagePart|MessagePart[]|Message $systemInstruction) void
759761
+getSystemInstruction() Message?
760-
+setGenerationConfig(GenerationConfig $config) void
761-
+getGenerationConfig() GenerationConfig?
762+
+setCandidateCount(int $candidateCount) void
763+
+getCandidateCount() int
764+
+setMaxTokens(int $maxTokens) void
765+
+getMaxTokens() int
766+
+setTemperature(float $temperature) void
767+
+getTemperature() float
768+
+setTopK(int $topK) void
769+
+getTopK() int
770+
+setTopP(float $topP) void
771+
+getTopP() float
772+
+setOutputMimeType(string $outputMimeType) void
773+
+getOutputMimeType() string
774+
+setOutputSchema(array< string, mixed > $outputSchema) void
775+
+getOutputSchema() array< string, mixed >
776+
+setCustomOption(string $key, mixed $value) void
777+
+getCustomOption(string $key) mixed
778+
+getCustomOptions() array< string, mixed >
762779
+setTools(Tool[] $tools) void
763780
+getTools() Tool[]
764781
+getJsonSchema() array< string, mixed >$
765782
}
766-
class GenerationConfig {
767-
+setValue(string $key, mixed $value) void
768-
+getValue(string $key) mixed
769-
+getValues() array< string, mixed >
770-
+getAdditionalValues() array< string, mixed >
771-
+getJsonSchema() array< string, mixed >$
772-
}
773-
class TextGenerationConfig {
774-
}
775-
class ImageGenerationConfig {
776-
}
777-
class TextToSpeechConversionConfig {
778-
}
779-
class SpeechGenerationConfig {
780-
}
781-
class EmbeddingGenerationConfig {
782-
}
783783
class Tool {
784784
+getType() ToolType
785785
+getFunctionDeclarations() FunctionDeclaration[]?
@@ -832,6 +832,7 @@ direction LR
832832
class AiOption {
833833
INPUT_MODALITIES
834834
OUTPUT_MODALITIES
835+
SYSTEM_INSTRUCTION
835836
CANDIDATE_COUNT
836837
MAX_TOKENS
837838
TEMPERATURE
@@ -868,7 +869,6 @@ direction LR
868869
<<interface>> HttpClient
869870
<<interface>> WithAuthentication
870871
<<interface>> Authentication
871-
<<interface>> GenerationConfig
872872
<<Enumeration>> AiCapability
873873
<<Enumeration>> AiOption
874874
<<Enumeration>> AiProviderType
@@ -888,7 +888,6 @@ direction LR
888888
AiModelMetadata "1" o-- "0..*" AiSupportedOption
889889
AiModelRequirements "1" o-- "1..*" AiCapability
890890
AiModelRequirements "1" o-- "0..*" AiRequiredOption
891-
AiModelConfig "1" o-- "0..1" GenerationConfig
892891
AiModelConfig "1" o-- "0..*" Tool
893892
Tool "1" o-- "0..*" FunctionDeclaration
894893
Tool "1" o-- "0..1" WebSearch
@@ -905,9 +904,4 @@ direction LR
905904
AiModel <|-- AiTextToSpeechConversionOperationModel
906905
AiModel <|-- AiSpeechGenerationOperationModel
907906
AiModel <|-- AiEmbeddingGenerationOperationModel
908-
GenerationConfig <|-- TextGenerationConfig
909-
GenerationConfig <|-- ImageGenerationConfig
910-
GenerationConfig <|-- TextToSpeechConversionConfig
911-
GenerationConfig <|-- SpeechGenerationConfig
912-
GenerationConfig <|-- EmbeddingGenerationConfig
913907
```

0 commit comments

Comments
 (0)