From b10ddf0673c586fc507acb0878fd88cd25da328a Mon Sep 17 00:00:00 2001 From: Adeel Hassan Date: Wed, 11 Dec 2024 16:44:50 -0500 Subject: [PATCH] update import paths for e84_geoai_common --- src/natural_language_geocoding/__init__.py | 3 ++- src/natural_language_geocoding/prompt.py | 2 +- src/natural_language_geocoding_demo/app.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/natural_language_geocoding/__init__.py b/src/natural_language_geocoding/__init__.py index 4617dc0..b3d1756 100644 --- a/src/natural_language_geocoding/__init__.py +++ b/src/natural_language_geocoding/__init__.py @@ -1,4 +1,5 @@ -from e84_geoai_common.llm import LLM, extract_data_from_text +from e84_geoai_common.llm.core import LLM +from e84_geoai_common.llm.extraction import extract_data_from_text from shapely.geometry.base import BaseGeometry from natural_language_geocoding.models import SpatialNode diff --git a/src/natural_language_geocoding/prompt.py b/src/natural_language_geocoding/prompt.py index ac2e235..2c6f702 100644 --- a/src/natural_language_geocoding/prompt.py +++ b/src/natural_language_geocoding/prompt.py @@ -1,7 +1,7 @@ import json import os from e84_geoai_common.util import singleline -from e84_geoai_common.llm import ExtractDataExample +from e84_geoai_common.llm.extraction import ExtractDataExample from natural_language_geocoding.models import SpatialNode diff --git a/src/natural_language_geocoding_demo/app.py b/src/natural_language_geocoding_demo/app.py index 140e1a0..b65c2db 100644 --- a/src/natural_language_geocoding_demo/app.py +++ b/src/natural_language_geocoding_demo/app.py @@ -1,4 +1,4 @@ -from e84_geoai_common.llm import BedrockClaudeLLM +from e84_geoai_common.llm.models import BedrockClaudeLLM from natural_language_geocoding import extract_geometry_from_text from e84_geoai_common.geometry import simplify_geometry, geometry_to_geojson from e84_geoai_common.debugging import display_geometry