python-oracledb/samples/vector_string.py at v2.2.1 · oracle/python-oracledb · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
# -----------------------------------------------------------------------------
# Copyright (c) 2023, Oracle and/or its affiliates.
#
# This software is dual-licensed to you under the Universal Permissive License
# (UPL) 1.0 as shown at https://oss.oracle.com/licenses/upl and Apache License
# 2.0 as shown at http://www.apache.org/licenses/LICENSE-2.0. You may choose
# either license.
#
# If you elect to accept the software under the Apache License, Version 2.0,
# the following applies:
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#    https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# -----------------------------------------------------------------------------

# -----------------------------------------------------------------------------
# vector_string.py
#
# Demonstrates how to use the Oracle Database 23ai VECTOR data type
# when using python-oracledb thick mode with pre-23ai Oracle Client libraries.
# See vector.py for a more efficient example for other deployments.
# -----------------------------------------------------------------------------

import sys
import oracledb
import sample_env

# Although the code in this example could run with python-oracledb thin mode,
# it is not efficient so it should only be used for thick mode with old Oracle
# Client libraries.
oracledb.init_oracle_client(lib_dir=sample_env.get_oracle_client())

connection = oracledb.connect(
    user=sample_env.get_main_user(),
    password=sample_env.get_main_password(),
    dsn=sample_env.get_connect_string(),
)

# this script only works with Oracle Database 23.4 or later
if sample_env.get_server_version() < (23, 4):
    sys.exit("This example requires Oracle Database 23.4 or later.")


with connection.cursor() as cursor:
    # Single-row insert
    vector1_data_32 = "[1.625, 1.5, 1.0]"
    vector1_data_64 = "[11.25, 11.75, 11.5]"
    vector1_data_8 = "[1, 2, 3]"

    cursor.execute(
        "insert into SampleVectorTab (v32, v64, v8) values (:1, :2, :3)",
        [vector1_data_32, vector1_data_64, vector1_data_8],
    )

    # Multi-row insert
    vector2_data_32 = "[2.625, 2.5, 2.0]"
    vector2_data_64 = "[22.25, 22.75, 22.5]"
    vector2_data_8 = "[4, 5, 6]"

    vector3_data_32 = "[3.625, 3.5, 3.0]"
    vector3_data_64 = "[33.25, 33.75, 33.5]"
    vector3_data_8 = "[7, 8, 9]"

    rows = [
        (vector2_data_32, vector2_data_64, vector2_data_8),
        (vector3_data_32, vector3_data_64, vector3_data_8),
    ]

    cursor.executemany(
        """insert into SampleVectorTab (v32, v64, v8)
                          values (:1, :2, :3)""",
        rows,
    )

    # Query
    cursor.execute("select * from SampleVectorTab")

    # Each vector is represented as a list
    for row in cursor:
        print(row)