update import paths for e84_geoai_common

Element84 · Dec 11, 2024 · b10ddf0 · b10ddf0
1 parent d0a0273
commit b10ddf0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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