Structured Data Generation with JSON Schema

from openai import OpenAI
import json
import os

schema= """{
  "title": "Client Profile",
  "type": "object",
  "properties": {
    "name": {
      "title": "Name",
      "type": "string",
      "maxLength": 100
    },
    "yearsOfActivity": {
      "title": "Years of Activity",
      "description": "Age of the company in years",
      "type": "integer",
      "minimum": 0
    },
    "previousPurchases": {
      "title": "Previous Purchases",
      "type": "array",
      "items": {
        "$ref": "#/definitions/purchase"
      }
    },
    "areasOfInterest": {
      "title": "Areas of Interest",
      "type": "array",
      "items": {
        "$ref": "#/definitions/interest"
      }
    },
    "lastPurchase": {
      "title": "Last Purchase",
      "description": "Reference of last product sold to the client",
      "type": "string",
      "maxLength": 60
    }
  },
  "required": ["name", "yearsOfActivity", "previousPurchases", "areasOfInterest", "lastPurchase"],
  "definitions": {
    "purchase": {
      "title": "Purchase",
      "type": "object",
      "properties": {
        "type": {
          "title": "Purchase Type",
          "description": "Indicates if the purchase was a single item or multiple items",
          "enum": ["Single", "Multiple"],
          "type": "string"
        },
        "details": {
          "title": "Purchase Details",
          "description": "Details about the purchase",
          "type": "string"
        }
      },
      "required": ["type", "details"]
    },
    "interest": {
      "title": "Interest",
      "description": "Enumeration of the areas of interest",
      "type": "string",
      "enum": ["AI", "HR", "Finance", "Clothing", "Engineering", "CSR"]
    }
  }
}
"""

chat_completion = client.chat.completions.create(
    model="alfred-40b-1123",
    messages=[
        {"role": "user", "content": "Give me a detailed client profile for a B2B solution."},
    ],
    temperature=1,
    max_tokens=400,
    stream=False,
    stop=["</s>"],
    extra_body={"json_schema": schema},
)
#print the client profile
print(json.loads(chat_completion.choices[0].message.content))

{'name': 'Tech-Enabled Manufacturing Co.',
'yearsOfActivity': 20,
'previousPurchases': [
    {'type': 'Single', 'details': 'Bought 20 enterprise-grade 3D printers last year'},
    {'type': 'Multiple', 'details': 'Purchased maintenance and repair services for heavy machinery multiple times over the past 5 years'}],
'areasOfInterest': ['AI', 'Clothing'],
'lastPurchase': 'Maintenance and repair services for heavy machinery, 6 month'}

import json

#Make sur that you initialized the client before

# Define the JSON Schema for extracting inquiry details
schema = """{
    "title": "Inquiry Details",
    "type": "object",
    "properties": {
        "name": {
            "type": "string"
        },
        "email": {
            "type": "string"
        },
        "inquiry_type": {
            "type": "string"
        },
        "summary": {
            "type": "string",
            "pattern": "The summary of the conversation is: (.+)"
        }
    },
    "required": ["name", "email", "inquiry_type", "summary"]
}"""

# Example text from which to extract data
text_to_extract_from = """
Here is an email :

Subject: Inquiry about Custom Solutions for E-Commerce Platforms

Content of the e-mail:

``
Dear Paradigm Support Team,

I hope this message finds you well. My name is John Doe, and I am the CTO of XYZ Retail, an online retail company specializing in bespoke home decor items. We have been exploring advanced AI solutions to enhance our customer experience and streamline our operations, and we came across your innovative product offerings.

We are particularly interested in understanding how your AI solutions can be integrated with our existing e-commerce platform to provide personalized product recommendations and optimize our supply chain management.

Could you provide us with more detailed information about the following:

1. The integration process with existing e-commerce platforms, specifically Shopify and Magento.
2. Case studies or examples of similar implementations in the retail sector.
3. The scalability of your solutions to accommodate our rapid growth.
4. The level of customer support and technical assistance provided during and post-integration.

Additionally, we would like to inquire about the possibility of a custom solution tailored to our specific needs, including predictive analytics for inventory management and AI-driven insights for market trend analysis.

I look forward to your detailed response and hope to explore a potential collaboration.

Best regards,

John Doe
CTO, XYZ Retail
Email: johndoe@example.com
Phone: +1234567890
``

Review the following message and extract the key details including the sender's name, email address, type of inquiry, and the main message content. Format the extracted information according to the predefined JSON schema, ensuring that each piece of information is accurately captured and categorized.

Now give me the structured json:
"""

# Extract data using the defined schema
response = client.chat.completions.create(
    model="alfred-40b-1123",
    messages=[
        {"role": "user", "content": text_to_extract_from},
    ],
    temperature=0.7,
    max_tokens=1000,
    stream=False,
    stop=["</s>"],
    extra_body={"json_schema": schema},
)

# Print the extracted structured data
print(json.loads(response.choices[0].message.content))

{
   "name":"John Doe",
   "email":"johndoe@example.com",
   "inquiry_type":"Request for information on custom AI solutions for e-commerce platforms",
   "summary":"The summary of the conversation is: John Doe, CTO of XYZ Retail, is interested in understanding how Paradigm's AI solutions can be integrated with their existing e-commerce platform (Shopify and Magento) to provide personalized product recommendations and optimize supply chain management. John has also inquired about the scalability of the solutions, level of customer support, and the possibility of a custom solution tailored to their specific needs, including predictive analytics for inventory management and AI-driven insights for market trend analysis."
}