Skip to content

Commit 6a2ba46

Browse files
committed
feat: + google gemini 2.0 flash exp (a very good model)
1 parent 5b027a9 commit 6a2ba46

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

chat_workflow/llm/providers/google.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def create_model(self, name: str, model: str, tools: Optional[List] = None, **kw
2020
return llm.bind_tools(tools) if tools else llm
2121

2222
def list_models(self) -> List[str]:
23-
return ["gemini-1.5-pro", "gemini-1.5-flash"]
23+
return ["gemini-2.0-flash-exp", "gemini-1.5-pro", "gemini-1.5-flash"]
2424

2525
@property
2626
def name(self) -> str:
@@ -29,6 +29,7 @@ def name(self) -> str:
2929
@property
3030
def capabilities(self) -> Dict[str, Set[ModelCapability]]:
3131
return {
32+
"gemini-2.0-flash-exp": {ModelCapability.TEXT_TO_TEXT, ModelCapability.IMAGE_TO_TEXT, ModelCapability.TOOL_CALLING},
3233
"gemini-1.5-pro": {ModelCapability.TEXT_TO_TEXT, ModelCapability.IMAGE_TO_TEXT, ModelCapability.TOOL_CALLING},
3334
"gemini-1.5-flash": {ModelCapability.TEXT_TO_TEXT, ModelCapability.IMAGE_TO_TEXT, ModelCapability.TOOL_CALLING},
3435
}

0 commit comments

Comments
 (0)