Commit dbf32f43 authored by YaningGao's avatar YaningGao
Browse files

minor

parent 23b461e6
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -10,7 +10,7 @@ from PIL import Image
import io
import requests as http_requests

from vagen.mllm_agent.model_interface.base_model import BaseModelInterface
from vagen.inference.model_interface.base_model import BaseModelInterface
from .model_config import ClaudeModelConfig

logger = logging.getLogger(__name__)
+1 −1
Original line number Diff line number Diff line
from dataclasses import dataclass, field
from typing import Optional, Dict, Any, List
from vagen.mllm_agent.model_interface.base_model_config import BaseModelConfig
from vagen.inference.model_interface.base_model_config import BaseModelConfig

@dataclass
class ClaudeModelConfig(BaseModelConfig):
+1 −1
Original line number Diff line number Diff line
@@ -8,7 +8,7 @@ import io
import google.generativeai as genai
from google.generativeai.types import HarmCategory, HarmBlockThreshold

from vagen.mllm_agent.model_interface.base_model import BaseModelInterface
from vagen.inference.model_interface.base_model import BaseModelInterface
from .model_config import GeminiModelConfig

logger = logging.getLogger(__name__)
+1 −1
Original line number Diff line number Diff line
from dataclasses import dataclass, field
from typing import Optional, Dict, Any, List
from vagen.mllm_agent.model_interface.base_model_config import BaseModelConfig
from vagen.inference.model_interface.base_model_config import BaseModelConfig

@dataclass
class GeminiModelConfig(BaseModelConfig):
+1 −1
Original line number Diff line number Diff line
@@ -8,7 +8,7 @@ from openai import OpenAI
from PIL import Image
import io

from vagen.mllm_agent.model_interface.base_model import BaseModelInterface
from vagen.inference.model_interface.base_model import BaseModelInterface
from .model_config import OpenAIModelConfig

logger = logging.getLogger(__name__)
Loading