missing url

Author: aembryonic

analysis_module/serializers.py

@@ -159,8 +159,8 @@ class EntryExtractionSerializer(serializers.Serializer):
 class EntryExtractionSerializerLLM(serializers.Serializer):
 
     documents = DocumentEntryExtractionUnionField()
-    af_id =  serializers.CharField()
-    project_id = serializers.CharField()
+    af_id =  serializers.IntegerField()
+    project_id = serializers.IntegerField()
     callback_url = serializers.CharField()
     request_type = serializers.ChoiceField(
         choices=ExtractionRequestTypeChoices,

analysis_module/utils.py

@@ -287,7 +287,7 @@ def get_ecs_id_param_name(request_type: NLPRequest.FeaturesType):
         NLPRequest.FeaturesType.TOPICMODEL: "topicmodel_id",
         NLPRequest.FeaturesType.GEOLOCATION: "geolocation_id",
         NLPRequest.FeaturesType.ENTRY_EXTRACTION: "entryextraction_id",
-        NLPRequest.FeaturesType.ENTRY_CLASSIFICATION_LLM: "entryextraction_llm_id",
+        NLPRequest.FeaturesType.ENTRY_EXTRACTION_LLM: "entryextraction_id",
         NLPRequest.FeaturesType.TEXT_EXTRACTION: "textextraction_id",
         NLPRequest.FeaturesType.SUMMARIZATION_V3: "summarization_id"
     }
@@ -299,7 +299,7 @@ def get_ecs_url(request_type: NLPRequest.FeaturesType):
         NLPRequest.FeaturesType.TOPICMODEL: urljoin(TOPICMODEL_ECS_ENDPOINT, "/get_excerpt_clusters"),
         NLPRequest.FeaturesType.GEOLOCATION: urljoin(GEOLOCATION_ECS_ENDPOINT, "/get_geolocations"),
         NLPRequest.FeaturesType.ENTRY_EXTRACTION: urljoin(ENTRYEXTRACTION_ECS_ENDPOINT, "/extract_entries"),
-        NLPRequest.FeaturesType.ENTRY_CLASSIFICATION_LLM: urljoin(ENTRYEXTRACTION_ECS_ENDPOINT, "/extract_entries_llm"),
+        NLPRequest.FeaturesType.ENTRY_EXTRACTION_LLM: urljoin(ENTRYEXTRACTION_ECS_ENDPOINT, "/extract_entries_llm"),
         NLPRequest.FeaturesType.TEXT_EXTRACTION: urljoin(TEXT_EXTRACTION_ECS_ENDPOINT, "/extract_document"),
         NLPRequest.FeaturesType.SUMMARIZATION_V3: urljoin(SUMMARIZATION_V3_ECS_ENDPOINT, "/generate_report")
     }

core_server/settings.py

@@ -56,7 +56,7 @@
 SUMMARIZATION_V3_ECS_ENDPOINT = env("SUMMARIZATION_V3_ECS_ENDPOINT")
 TEXT_EXTRACTION_ECS_ENDPOINT = env("TEXTEXTRACTION_ECS_ENDPOINT")
 ENTRYEXTRACTION_ECS_ENDPOINT = env("ENTRYEXTRACTION_ECS_ENDPOINT")
-ENTRYEXTRACTION_ECS_ENDPOINT = env("ENTRYEXTRACTION_LLM_ECS_ENDPOINT")
+ENTRYEXTRACTION_LLM_ECS_ENDPOINT = env("ENTRYEXTRACTION_LLM_ECS_ENDPOINT")
 GEOLOCATION_ECS_ENDPOINT = env("GEOLOCATION_ECS_ENDPOINT")
 TOPICMODEL_ECS_ENDPOINT = env("TOPICMODEL_ECS_ENDPOINT")
 

core_server/urls.py

@@ -31,6 +31,7 @@
 )
 
 from analysis_module.views.predictions_v2 import entry_classification as llm_entry_classification
+from analysis_module.views.entry_extraction_v2 import entry_extraction as llm_entry_extraction
 from analysis_module.views.entry_extraction import (
     entry_extraction
 )
@@ -44,6 +45,7 @@
     path("api/v1/summarization/", summarization),
     path("api/v1/text-extraction/", text_extraction),
     path("api/v1/entry-extraction-classification/", entry_extraction),
+    path("api/v1/entry-extraction-classification-llm/", llm_entry_extraction),
     path("api/v1/ngrams/", ngrams),
     path("api/v1/geolocation/", geolocation),
     path("api/v1/tags-mapping/", tags_mapping),