fix: update structure

Author: FlorentLvr

src/custom/modules/opendata/pipelines/PerplexityOrganizationAnalysisPipeline.py

@@ -553,13 +553,13 @@ def add_corporate_structure(
 
         # Add strategic alliances to graph
         logger.info(f"---> Mapping organization strategic alliances")
-        organization_strategic_alliances = response["organization_strategic_alliances"]
-        organization_partnerships = organization_strategic_alliances["partnerships"]
-        organization_joint_ventures = organization_strategic_alliances["joint_ventures"]
-        organization_marketing_alliances = organization_strategic_alliances["marketing_alliances"]
-        organization_research_collaborations = organization_strategic_alliances["research_collaborations"]
-        organization_technology_licensing = organization_strategic_alliances["technology_licensing"]
-        organization_distribution_agreements = organization_strategic_alliances["distribution_agreements"]
+        organization_strategic_alliances = response.get("organization_strategic_alliances", {})
+        organization_partnerships = organization_strategic_alliances.get("partnerships", [])
+        organization_joint_ventures = organization_strategic_alliances.get("joint_ventures", [])
+        organization_marketing_alliances = organization_strategic_alliances.get("marketing_alliances", [])
+        organization_research_collaborations = organization_strategic_alliances.get("research_collaborations", [])
+        organization_technology_licensing = organization_strategic_alliances.get("technology_licensing", [])
+        organization_distribution_agreements = organization_strategic_alliances.get("distribution_agreements", [])
 
         # Add strategic alliances to graph
         def add_strategic_alliances(

src/custom/modules/opendata/tests/pipelines/test_PerplexityOrganizationAnalysisPipeline.py

@@ -2,8 +2,8 @@
 from src.core.integrations.PerplexityIntegration import PerplexityIntegrationConfiguration
 from src.core.integrations.NaasIntegration import NaasIntegrationConfiguration
 from src.core.workflows.abi.CreateOntologyYAML import CreateOntologyYAMLConfiguration
-from src.custom.pipelines.opendata.PerplexityOrganizationAnalysisPipeline import PerplexityOrganizationAnalysisPipeline, PerplexityOrganizationAnalysisPipelineConfiguration, PerplexityOrganizationAnalysisPipelineParameters
-from src.custom.workflows.opendata.PerplexityGetOrganizationWorkflows import PerplexityOrganizationWorkflowsConfiguration
+from src.custom.modules.opendata.pipelines.PerplexityOrganizationAnalysisPipeline import PerplexityOrganizationAnalysisPipeline, PerplexityOrganizationAnalysisPipelineConfiguration, PerplexityOrganizationAnalysisPipelineParameters
+from src.custom.modules.opendata.workflows.PerplexityGetOrganizationWorkflows import PerplexityOrganizationWorkflowsConfiguration
 
 # Initialize ontology store
 ontology_store = services.ontology_store_service
@@ -28,8 +28,8 @@
     create_ontology_yaml_config=create_ontology_yaml_config,
 ))
 
-organization_name = "Softbank"
-website = "https://www.softbank.jp/"
+organization_name = "Michelin"
+website = "https://www.michelin.com/"
 
 pipeline.run(PerplexityOrganizationAnalysisPipelineParameters(
     organization_name=organization_name,

src/custom/modules/opendata/workflows/PerplexityGetOrganizationWorkflows.py

@@ -93,7 +93,7 @@ def extract_data_from_perplexity(
             system_prompt = """You are an experimented OSINT investigator. 
             Your goal is to extract information about a given organization.
             Rules: 
-            1. Provide only the final answer. It is important that you do not include any explanation on the steps below. 
+            1. Provide only the final answer in JSON format as follow: ```json <JSON>```
             2. Do not show the intermediate steps information.
             3. Be precise and exhaustive.
             4. Do not include any information that is not related to the question.