diff --git a/auto-claude/integrations/graphiti/providers_pkg/embedder_providers/google_embedder.py b/auto-claude/integrations/graphiti/providers_pkg/embedder_providers/google_embedder.py index 8656cb33..338f479e 100644 --- a/auto-claude/integrations/graphiti/providers_pkg/embedder_providers/google_embedder.py +++ b/auto-claude/integrations/graphiti/providers_pkg/embedder_providers/google_embedder.py @@ -8,11 +8,11 @@ Uses the google-generativeai SDK for text embeddings. from typing import TYPE_CHECKING, Any +from ..exceptions import ProviderError, ProviderNotInstalled + if TYPE_CHECKING: from graphiti_config import GraphitiConfig -from ..exceptions import ProviderError, ProviderNotInstalled - # Default embedding model for Google DEFAULT_GOOGLE_EMBEDDING_MODEL = "text-embedding-004" diff --git a/auto-claude/integrations/graphiti/providers_pkg/llm_providers/google_llm.py b/auto-claude/integrations/graphiti/providers_pkg/llm_providers/google_llm.py index 55fc84bd..30facc36 100644 --- a/auto-claude/integrations/graphiti/providers_pkg/llm_providers/google_llm.py +++ b/auto-claude/integrations/graphiti/providers_pkg/llm_providers/google_llm.py @@ -8,11 +8,11 @@ Uses the google-generativeai SDK. from typing import TYPE_CHECKING, Any +from ..exceptions import ProviderError, ProviderNotInstalled + if TYPE_CHECKING: from graphiti_config import GraphitiConfig -from ..exceptions import ProviderError, ProviderNotInstalled - # Default model for Google LLM DEFAULT_GOOGLE_LLM_MODEL = "gemini-2.0-flash"