fixed import names errors of assistants

uhh-lt · Feb 27, 2025 · 15536d6 · 15536d6
1 parent 94ef68c
commit 15536d6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/src/fundus_murag/assistant/gemini_fundus_assistant.py b/src/fundus_murag/assistant/gemini_fundus_assistant.py
@@ -3,7 +3,6 @@
 import google.generativeai as genai
 import pandas as pd
 import vertexai
-from base_fundus_assistant import BaseFundusAssistant
 from google.oauth2.service_account import Credentials
 from loguru import logger
 from vertexai.generative_models import (
@@ -13,6 +12,7 @@
     GenerativeModel,
 )
 
+from fundus_murag.assistant.base_fundus_assistant import BaseFundusAssistant
 from fundus_murag.assistant.function_call_handler import FunctionCallHandler
 from fundus_murag.assistant.prompt import ASSISTANT_SYSTEM_INSTRUCTION
 from fundus_murag.assistant.tools import FUNDUS_TOOL

diff --git a/src/fundus_murag/assistant/openai_fundus_assistant.py b/src/fundus_murag/assistant/openai_fundus_assistant.py
@@ -2,9 +2,9 @@
 
 import openai
 import pandas as pd
-from base_fundus_assistant import BaseFundusAssistant
 from loguru import logger
 
+from fundus_murag.assistant.base_fundus_assistant import BaseFundusAssistant
 from fundus_murag.assistant.prompt import ASSISTANT_SYSTEM_INSTRUCTION
 from fundus_murag.config.config import load_config
 from fundus_murag.singleton_meta import SingletonMeta

diff --git a/src/fundus_murag/assistant/openai_image_analysis_assistant.py b/src/fundus_murag/assistant/openai_image_analysis_assistant.py
@@ -1,9 +1,11 @@
 from typing import Literal, TypedDict
 
 import openai
-from base_image_analysis_assistant import BaseImageAnalysisAssistant
 from loguru import logger
 
+from fundus_murag.assistant.base_image_analysis_assistant import (
+    BaseImageAnalysisAssistant,
+)
 from fundus_murag.assistant.prompt import (
     IMAGE_ANALYSIS_IC_SYSTEM_INSTRUCTION,
     IMAGE_ANALYSIS_VQA_SYSTEM_INSTRUCTION,