Skip to content

Commit 69bdaa5

Browse files
committed
chore: added a constant for the metadata definition in the openai plugin
1 parent 2fbbb18 commit 69bdaa5

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

py/plugins/compat-oai/src/genkit/plugins/compat_oai/openai_plugin.py

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616

1717

1818
"""OpenAI OpenAI API Compatible Plugin for Genkit."""
19+
from typing import Any
1920

2021
from openai import Client, OpenAI as OpenAIClient
2122

@@ -32,6 +33,14 @@
3233

3334
OPENAI_PLUGIN_NAME = 'openai'
3435

36+
def default_openai_metadata(name: str) -> dict[str, Any]:
37+
return {
38+
'model': {
39+
'label': f"OpenAI - {name}",
40+
'supports': {'multiturn': True}
41+
},
42+
}
43+
3544
class OpenAI(Plugin):
3645
"""A plugin for integrating OpenAI compatible models with the Genkit framework.
3746
@@ -102,12 +111,7 @@ def _define_openai_model(self, ai: GenkitRegistry, name: str) -> None:
102111
name=f'openai/{name}',
103112
fn=handler,
104113
config_schema=OpenAIConfig,
105-
metadata={
106-
'model': {
107-
'label': f"OpenAI - {name}",
108-
'supports': {'multiturn': True}
109-
},
110-
},
114+
metadata=default_openai_metadata(name)
111115
)
112116

113117

0 commit comments

Comments
 (0)