Skip to content

Commit af31af4

Browse files
committed
feat: add dynamic provider static factories
1 parent 4f0bf5e commit af31af4

1 file changed

Lines changed: 133 additions & 0 deletions

File tree

src/main/java/io/weaviate/client6/v1/api/collections/generate/DynamicProvider.java

Lines changed: 133 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,18 @@
33
import java.util.function.Function;
44

55
import io.weaviate.client6.v1.api.collections.generative.AnthropicGenerative;
6+
import io.weaviate.client6.v1.api.collections.generative.AnyscaleGenerative;
7+
import io.weaviate.client6.v1.api.collections.generative.AwsGenerative;
8+
import io.weaviate.client6.v1.api.collections.generative.AzureOpenAiGenerative;
9+
import io.weaviate.client6.v1.api.collections.generative.CohereGenerative;
10+
import io.weaviate.client6.v1.api.collections.generative.DatabricksGenerative;
11+
import io.weaviate.client6.v1.api.collections.generative.FriendliaiGenerative;
12+
import io.weaviate.client6.v1.api.collections.generative.GoogleGenerative;
13+
import io.weaviate.client6.v1.api.collections.generative.MistralGenerative;
14+
import io.weaviate.client6.v1.api.collections.generative.NvidiaGenerative;
15+
import io.weaviate.client6.v1.api.collections.generative.OllamaGenerative;
16+
import io.weaviate.client6.v1.api.collections.generative.OpenAiGenerative;
17+
import io.weaviate.client6.v1.api.collections.generative.XaiGenerative;
618
import io.weaviate.client6.v1.internal.ObjectBuilder;
719
import io.weaviate.client6.v1.internal.grpc.protocol.WeaviateProtoGenerative;
820

@@ -18,4 +30,125 @@ public static DynamicProvider anthropic(
1830
Function<AnthropicGenerative.Provider.Builder, ObjectBuilder<AnthropicGenerative.Provider>> fn) {
1931
return AnthropicGenerative.Provider.of(fn);
2032
}
33+
34+
/**
35+
* Configure {@code generative-anyscale} as a dynamic provider.
36+
*
37+
* @param fn Lambda expression for optional parameters.
38+
*/
39+
public static DynamicProvider anyscale(
40+
Function<AnyscaleGenerative.Provider.Builder, ObjectBuilder<AnyscaleGenerative.Provider>> fn) {
41+
return AnyscaleGenerative.Provider.of(fn);
42+
}
43+
44+
/**
45+
* Configure {@code generative-aws} as a dynamic provider.
46+
*
47+
* @param fn Lambda expression for optional parameters.
48+
*/
49+
public static DynamicProvider aws(
50+
Function<AwsGenerative.Provider.Builder, ObjectBuilder<AwsGenerative.Provider>> fn) {
51+
return AwsGenerative.Provider.of(fn);
52+
}
53+
54+
/**
55+
* Configure {@code generative-cohere} as a dynamic provider.
56+
*
57+
* @param fn Lambda expression for optional parameters.
58+
*/
59+
public static DynamicProvider cohere(
60+
Function<CohereGenerative.Provider.Builder, ObjectBuilder<CohereGenerative.Provider>> fn) {
61+
return CohereGenerative.Provider.of(fn);
62+
}
63+
64+
/**
65+
* Configure {@code generative-databricks} as a dynamic provider.
66+
*
67+
* @param fn Lambda expression for optional parameters.
68+
*/
69+
public static DynamicProvider databricks(
70+
Function<DatabricksGenerative.Provider.Builder, ObjectBuilder<DatabricksGenerative.Provider>> fn) {
71+
return DatabricksGenerative.Provider.of(fn);
72+
}
73+
74+
/**
75+
* Configure {@code generative-friendliai} as a dynamic provider.
76+
*
77+
* @param fn Lambda expression for optional parameters.
78+
*/
79+
public static DynamicProvider friendliai(
80+
Function<FriendliaiGenerative.Provider.Builder, ObjectBuilder<FriendliaiGenerative.Provider>> fn) {
81+
return FriendliaiGenerative.Provider.of(fn);
82+
}
83+
84+
/**
85+
* Configure {@code generative-palm} as a dynamic provider.
86+
*
87+
* @param fn Lambda expression for optional parameters.
88+
*/
89+
public static DynamicProvider google(
90+
Function<GoogleGenerative.Provider.Builder, ObjectBuilder<GoogleGenerative.Provider>> fn) {
91+
return GoogleGenerative.Provider.of(fn);
92+
}
93+
94+
/**
95+
* Configure {@code generative-mistral} as a dynamic provider.
96+
*
97+
* @param fn Lambda expression for optional parameters.
98+
*/
99+
public static DynamicProvider mistral(
100+
Function<MistralGenerative.Provider.Builder, ObjectBuilder<MistralGenerative.Provider>> fn) {
101+
return MistralGenerative.Provider.of(fn);
102+
}
103+
104+
/**
105+
* Configure {@code generative-nvidia} as a dynamic provider.
106+
*
107+
* @param fn Lambda expression for optional parameters.
108+
*/
109+
public static DynamicProvider nvidia(
110+
Function<NvidiaGenerative.Provider.Builder, ObjectBuilder<NvidiaGenerative.Provider>> fn) {
111+
return NvidiaGenerative.Provider.of(fn);
112+
}
113+
114+
/**
115+
* Configure {@code generative-ollama} as a dynamic provider.
116+
*
117+
* @param fn Lambda expression for optional parameters.
118+
*/
119+
public static DynamicProvider ollama(
120+
Function<OllamaGenerative.Provider.Builder, ObjectBuilder<OllamaGenerative.Provider>> fn) {
121+
return OllamaGenerative.Provider.of(fn);
122+
}
123+
124+
/**
125+
* Configure {@code generative-openai} as a dynamic provider.
126+
*
127+
* @param fn Lambda expression for optional parameters.
128+
*/
129+
public static DynamicProvider openai(
130+
Function<OpenAiGenerative.Provider.Builder, ObjectBuilder<OpenAiGenerative.Provider>> fn) {
131+
return OpenAiGenerative.Provider.of(fn);
132+
}
133+
134+
/**
135+
* Configure {@code generative-openai} as a dynamic provider.
136+
*
137+
* @param fn Lambda expression for optional parameters.
138+
*/
139+
public static DynamicProvider azure(
140+
Function<AzureOpenAiGenerative.Provider.Builder, ObjectBuilder<AzureOpenAiGenerative.Provider>> fn) {
141+
return AzureOpenAiGenerative.Provider.of(fn);
142+
}
143+
144+
/**
145+
* Configure {@code generative-xai} as a dynamic provider.
146+
*
147+
* @param fn Lambda expression for optional parameters.
148+
*/
149+
public static DynamicProvider xai(
150+
Function<XaiGenerative.Provider.Builder, ObjectBuilder<XaiGenerative.Provider>> fn) {
151+
return XaiGenerative.Provider.of(fn);
152+
}
153+
21154
}

0 commit comments

Comments
 (0)