enhance gemini models

langgenius · Dec 9, 2024 · f1dce52 · f1dce52
1 parent 93ab6f3
commit f1dce52
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/api/core/model_runtime/model_providers/google/llm/llm.py b/api/core/model_runtime/model_providers/google/llm/llm.py
@@ -1,6 +1,6 @@
 import base64
-import os
 import json
+import os
 import tempfile
 import time
 from collections.abc import Generator
@@ -10,21 +10,21 @@
 import google.generativeai as genai
 import requests
 from google.api_core import exceptions
-from google.generativeai.types import ContentType, GenerateContentResponse, File
+from google.generativeai.types import ContentType, File, GenerateContentResponse
 from google.generativeai.types.content_types import to_part
 
 from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta
 from core.model_runtime.entities.message_entities import (
     AssistantPromptMessage,
     DocumentPromptMessageContent,
     ImagePromptMessageContent,
-    VideoPromptMessageContent,
     PromptMessage,
     PromptMessageContentType,
     PromptMessageTool,
     SystemPromptMessage,
     ToolPromptMessage,
     UserPromptMessage,
+    VideoPromptMessageContent,
 )
 from core.model_runtime.errors.invoke import (
     InvokeAuthorizationError,