refactor: update import paths for find_provider and Message

This commit is contained in:
2024-10-31 11:58:05 -04:00
parent 2bed7221b3
commit c28e2a3839
2 changed files with 2 additions and 2 deletions
+1 -1
View File
@@ -5,7 +5,7 @@ from typing import Any, Dict, List, Literal, Optional
from pydantic import BaseModel, Field
from .providers import find_provider
from .utils import find_provider
MESSAGE_ROLE = Literal["system", "user", "assistant"]
+1 -1
View File
@@ -5,7 +5,6 @@ import google.generativeai as genai
import instructor
from pydantic import BaseModel
from ..models import Conversation, Message
from ..settings import settings
from ._base import BaseProvider
@@ -38,6 +37,7 @@ class Gemini(BaseProvider):
def send_conversation(self, conversation: "Conversation") -> "Message":
"""Send a conversation to the Gemini API."""
from ..models import Message
messages = [
{