Skip to content

Commit 676f346

Browse files
committed
feat: basic commit
1 parent 7b9b92c commit 676f346

File tree

1 file changed

+0
-40
lines changed

1 file changed

+0
-40
lines changed

py/plugins/vertex-ai/src/genkit/plugins/vertex_ai/model_garden/openai_compatiblility/openai_compatibility.py

-40
Original file line numberDiff line numberDiff line change
@@ -127,43 +127,3 @@ def to_openai_role(role: Role) -> ChatCompletionRole:
127127
return ChatCompletionRole.TOOL
128128
case _:
129129
raise ValueError(f"Role '{role}' doesn't map to an OpenAI role.")
130-
131-
132-
133-
class OllamaVersion(StrEnum):
134-
"""Available versions of the llama model.
135-
136-
This enum defines the available versions of the llama model that
137-
can be used through Vertex AI.
138-
"""
139-
LLAMA_3_1 = 'llama-3.1'
140-
LLAMA_3_2 = 'llama-3.2'
141-
LLAMA3_405_B = 'llama3-405b'
142-
143-
144-
SUPPORTED_MODELS = {
145-
OllamaVersion.LLAMA_3_1: ModelInfo(
146-
versions=['meta/llama3-405b-instruct-maas'],
147-
label='Llama 3.1',
148-
supports=Supports(
149-
multiturn=True, media=False, tools=True,
150-
systemRole=True, output=['text', 'json']
151-
)
152-
),
153-
OllamaVersion.LLAMA_3_2: ModelInfo(
154-
versions=['meta/llama-3.2-90b-vision-instruct-maas'],
155-
label='Llama 3.2',
156-
supports=Supports(
157-
multiturn=True, media=True, tools=True,
158-
systemRole=True, output=['text', 'json']
159-
)
160-
),
161-
OllamaVersion.LLAMA3_405_B: ModelInfo(
162-
versions=[],
163-
label='Llama 3.1 405b',
164-
supports=Supports(
165-
multiturn=True, media=False, tools=True,
166-
systemRole=True, output=['text']
167-
)
168-
)
169-
}

0 commit comments

Comments
 (0)