API Basis Pengetahuan

# Kode contoh ini hanya untuk referensi. Jangan gunakan langsung dalam lingkungan produksi.
import os

from alibabacloud_bailian20231229 import models as bailian_20231229_models
from alibabacloud_bailian20231229.client import Client as bailian20231229Client
from alibabacloud_tea_openapi import models as open_api_models
from alibabacloud_tea_util import models as util_models
from alibabacloud_tea_util.client import Client as UtilClient

def check_environment_variables():
    """Memeriksa variabel lingkungan yang diperlukan dan melaporkan yang hilang."""
    required_vars = {
        'ALIBABA_CLOUD_ACCESS_KEY_ID': 'Alibaba Cloud AccessKey ID Anda',
        'ALIBABA_CLOUD_ACCESS_KEY_SECRET': 'Rahasia AccessKey Alibaba Cloud Anda',
        'WORKSPACE_ID': 'ID ruang kerja Model Studio Anda'
    }
    missing_vars = []
    for var, description in required_vars.items():
        if not os.environ.get(var):
            missing_vars.append(var)
            print(f"Error: Atur variabel lingkungan {var} ({description}).")

    return len(missing_vars) == 0

def create_client() -> bailian20231229Client:
    """
    Membuat dan mengonfigurasi klien.

    Returns:
        bailian20231229Client: Klien yang telah dikonfigurasi.
    """
    config = open_api_models.Config(
        access_key_id=os.environ.get('ALIBABA_CLOUD_ACCESS_KEY_ID'),
        access_key_secret=os.environ.get('ALIBABA_CLOUD_ACCESS_KEY_SECRET')
    )
        # Ini adalah titik akhir cloud publik. Anda dapat mengubahnya sesuai kebutuhan.
    config.endpoint = 'bailian.ap-southeast-1.aliyuncs.com'
    return bailian20231229Client(config)

def retrieve_index(client, workspace_id, index_id, query):
    """
    Mengambil informasi dari basis pengetahuan tertentu.

    Args:
        client (bailian20231229Client): Klien.
        workspace_id (str): ID ruang kerja.
        index_id (str): ID basis pengetahuan.
        query (str): Kueri pencarian.

    Returns:
        Respons dari layanan Model Studio.
    """
    headers = {}
    retrieve_request = bailian_20231229_models.RetrieveRequest(
        index_id=index_id,
        query=query
    )
    runtime = util_models.RuntimeOptions()
    return client.retrieve_with_options(workspace_id, retrieve_request, headers, runtime)

def main():
    """
    Menggunakan layanan Model Studio untuk mengambil dari basis pengetahuan.

    Returns:
        str atau None: Segmen teks yang diambil jika panggilan berhasil. Jika tidak, None dikembalikan.
    """
    if not check_environment_variables():
        print("Validasi variabel lingkungan gagal.")
        return
    try:
        print("Langkah 1: Membuat klien.")
        client = create_client()
        print("Langkah 2: Mengambil dari basis pengetahuan.")
        index_id = input("Masukkan ID basis pengetahuan: ")  # Ini adalah nilai Data.Id yang dikembalikan oleh operasi CreateIndex. Anda juga dapat memperolehnya dari halaman basis pengetahuan di konsol Model Studio.
        query = input("Masukkan kueri pencarian: ")
        workspace_id = os.environ.get('WORKSPACE_ID')
        resp = retrieve_index(client, workspace_id, index_id, query)
        result = UtilClient.to_jsonstring(resp.body)
        print(result)
    except Exception as e:
        print(f"Terjadi kesalahan: {e}")
        return None

if __name__ == '__main__':
    main()

# Kode contoh ini hanya untuk referensi. Jangan menggunakannya di lingkungan produksi.
import hashlib
import os
import time

import requests
from alibabacloud_bailian20231229 import models as bailian_20231229_models
from alibabacloud_bailian20231229.client import Client as bailian20231229Client
from alibabacloud_tea_openapi import models as open_api_models
from alibabacloud_tea_util import models as util_models

def check_environment_variables():
    """Memeriksa variabel lingkungan yang diperlukan dan memberikan prompt untuk variabel yang hilang."""
    required_vars = {
        'ALIBABA_CLOUD_ACCESS_KEY_ID': 'ID AccessKey Alibaba Cloud',
        'ALIBABA_CLOUD_ACCESS_KEY_SECRET': 'Rahasia AccessKey Alibaba Cloud',
        'WORKSPACE_ID': 'ID ruang kerja Model Studio'
    }
    missing_vars = []
    for var, description in required_vars.items():
        if not os.environ.get(var):
            missing_vars.append(var)
            print(f"Error: Harap atur variabel lingkungan {var} ({description}).")

    return len(missing_vars) == 0

# Membuat klien.
def create_client() -> bailian20231229Client:
    """
    Membuat dan mengonfigurasi klien.

    Returns:
        bailian20231229Client: Klien yang telah dikonfigurasi.
    """
    config = open_api_models.Config(
        access_key_id=os.environ.get('ALIBABA_CLOUD_ACCESS_KEY_ID'),
        access_key_secret=os.environ.get('ALIBABA_CLOUD_ACCESS_KEY_SECRET')
    )
    # Ini adalah titik akhir publik. Anda dapat mengubahnya sesuai kebutuhan.
    config.endpoint = 'bailian.ap-southeast-1.aliyuncs.com'
    return bailian20231229Client(config)

def calculate_md5(file_path: str) -> str:
    """
    Menghitung hash MD5 dari file.

    Args:
        file_path (str): Jalur lokal file.

    Returns:
        str: Hash MD5 dari file.
    """
    md5_hash = hashlib.md5()

    # Membaca file dalam mode biner.
    with open(file_path, "rb") as f:
        # Membaca file dalam chunk untuk menghindari penggunaan memori tinggi untuk file besar.
        for chunk in iter(lambda: f.read(4096), b""):
            md5_hash.update(chunk)

    return md5_hash.hexdigest()

def get_file_size(file_path: str) -> int:
    """
    Mendapatkan ukuran file dalam byte.
    Args:
        file_path (str): Jalur lokal file.
    Returns:
        int: Ukuran file dalam byte.
    """
    return os.path.getsize(file_path)

# Mengajukan sewa unggah file.
def apply_lease(client, category_id, file_name, file_md5, file_size, workspace_id):
    """
    Mengajukan sewa unggah file dari layanan Model Studio.

    Args:
        client (bailian20231229Client): Klien.
        category_id (str): ID kategori.
        file_name (str): Nama file.
        file_md5 (str): Hash MD5 file.
        file_size (int): Ukuran file dalam byte.
        workspace_id (str): ID ruang kerja.

    Returns:
        Tanggapan dari layanan Model Studio.
    """
    headers = {}
    request = bailian_20231229_models.ApplyFileUploadLeaseRequest(
        file_name=file_name,
        md_5=file_md5,
        size_in_bytes=file_size,
    )
    runtime = util_models.RuntimeOptions()
    return client.apply_file_upload_lease_with_options(category_id, workspace_id, request, headers, runtime)

# Mengunggah file ke penyimpanan sementara.
def upload_file(pre_signed_url, headers, file_path):
    """
    Mengunggah file ke layanan Model Studio.
    Args:
        pre_signed_url (str): URL dari sewa unggah.
        headers (dict): Header permintaan untuk unggah.
        file_path (str): Jalur lokal file.
    """
    with open(file_path, 'rb') as f:
        file_content = f.read()
    upload_headers = {
        "X-bailian-extra": headers["X-bailian-extra"],
        "Content-Type": headers["Content-Type"]
    }
    response = requests.put(pre_signed_url, data=file_content, headers=upload_headers)
    response.raise_for_status()

# Menambahkan file ke kategori.
def add_file(client: bailian20231229Client, lease_id: str, parser: str, category_id: str, workspace_id: str):
    """
    Menambahkan file ke kategori tertentu dalam layanan Model Studio.

    Args:
        client (bailian20231229Client): Klien.
        lease_id (str): ID sewa.
        parser (str): Parser yang digunakan untuk file.
        category_id (str): ID kategori.
        workspace_id (str): ID ruang kerja.

    Returns:
        Tanggapan dari layanan Model Studio.
    """
    headers = {}
    request = bailian_20231229_models.AddFileRequest(
        lease_id=lease_id,
        parser=parser,
        category_id=category_id,
    )
    runtime = util_models.RuntimeOptions()
    return client.add_file_with_options(workspace_id, request, headers, runtime)

# Menanyakan status penguraian file.
def describe_file(client, workspace_id, file_id):
    """
    Menanyakan status penguraian file.

    Args:
        client (bailian20231229Client): Klien.
        workspace_id (str): ID ruang kerja.
        file_id (str): ID file.

    Returns:
        Tanggapan dari layanan Model Studio.
    """
    headers = {}
    runtime = util_models.RuntimeOptions()
    return client.describe_file_with_options(workspace_id, file_id, headers, runtime)

# Mengirim pekerjaan untuk menambahkan dokumen.
def submit_index_add_documents_job(client, workspace_id, index_id, file_id, source_type):
    """
    Menambahkan file yang telah diurai ke basis pengetahuan.

    Args:
        client (bailian20231229Client): Klien.
        workspace_id (str): ID ruang kerja.
        index_id (str): ID basis pengetahuan.
        file_id (str): ID file.
        source_type(str): Tipe data.

    Returns:
        Tanggapan dari layanan Model Studio.
    """
    headers = {}
    submit_index_add_documents_job_request = bailian_20231229_models.SubmitIndexAddDocumentsJobRequest(
        index_id=index_id,
        document_ids=[file_id],
        source_type=source_type
    )
    runtime = util_models.RuntimeOptions()
    return client.submit_index_add_documents_job_with_options(workspace_id, submit_index_add_documents_job_request,
                                                              headers, runtime)

# Menunggu pekerjaan penambahan dokumen selesai.
def get_index_job_status(client, workspace_id, job_id, index_id):
    """
    Menanyakan status pekerjaan indeks.

    Args:
        client (bailian20231229Client): Klien.
        workspace_id (str): ID ruang kerja.
        index_id (str): ID basis pengetahuan.
        job_id (str): ID pekerjaan.

    Returns:
        Tanggapan dari layanan Model Studio.
    """
    headers = {}
    get_index_job_status_request = bailian_20231229_models.GetIndexJobStatusRequest(
        index_id=index_id,
        job_id=job_id
    )
    runtime = util_models.RuntimeOptions()
    return client.get_index_job_status_with_options(workspace_id, get_index_job_status_request, headers, runtime)

# Menghapus file lama.
def delete_index_document(client, workspace_id, index_id, file_id):
    """
    Menghapus satu atau beberapa file secara permanen dari basis pengetahuan tertentu.

    Args:
        client (bailian20231229Client): Klien.
        workspace_id (str): ID ruang kerja.
        index_id (str): ID basis pengetahuan.
        file_id (str): ID file.

    Returns:
        Tanggapan dari layanan Model Studio.
    """
    headers = {}
    delete_index_document_request = bailian_20231229_models.DeleteIndexDocumentRequest(
        index_id=index_id,
        document_ids=[file_id]
    )
    runtime = util_models.RuntimeOptions()
    return client.delete_index_document_with_options(workspace_id, delete_index_document_request, headers, runtime)

def update_knowledge_base(
        file_path: str,
        workspace_id: str,
        index_id: str,
        old_file_id: str
):
    """
    Memperbarui basis pengetahuan di Model Studio.
    Args:
        file_path (str): Jalur lokal file yang diperbarui.
        workspace_id (str): ID ruang kerja.
        index_id (str): ID basis pengetahuan yang akan diperbarui.
        old_file_id (str): FileId dari file yang akan diganti.
    Returns:
        str atau None: ID basis pengetahuan jika berhasil, jika tidak maka None.
    """
    # Mengatur nilai default.
    category_id = 'default'
    parser = 'DASHSCOPE_DOCMIND'
    source_type = 'DATA_CENTER_FILE'
    try:
        # Langkah 1: Membuat klien.
        print("Langkah 1: Membuat klien.")
        client = create_client()
        # Langkah 2: Menyiapkan informasi file.
        print("Langkah 2: Menyiapkan informasi file.")
        file_name = os.path.basename(file_path)
        file_md5 = calculate_md5(file_path)
        file_size = get_file_size(file_path)
        # Langkah 3: Mengajukan sewa unggah.
        print("Langkah 3: Mengajukan sewa unggah dari Model Studio.")
        lease_response = apply_lease(client, category_id, file_name, file_md5, file_size, workspace_id)
        lease_id = lease_response.body.data.file_upload_lease_id
        upload_url = lease_response.body.data.param.url
        upload_headers = lease_response.body.data.param.headers
        # Langkah 4: Mengunggah file ke penyimpanan sementara.
        print("Langkah 4: Mengunggah file ke penyimpanan sementara.")
        upload_file(upload_url, upload_headers, file_path)
        # Langkah 5: Menambahkan file ke kategori.
        print("Langkah 5: Menambahkan file ke kategori.")
        add_response = add_file(client, lease_id, parser, category_id, workspace_id)
        file_id = add_response.body.data.file_id
        # Langkah 6: Memeriksa status file.
        print("Langkah 6: Memeriksa status file di Model Studio.")
        while True:
            describe_response = describe_file(client, workspace_id, file_id)
            status = describe_response.body.data.status
            print(f"Status file saat ini: {status}")
            if status == 'INIT':
                print("File sedang menunggu untuk diurai. Harap tunggu...")
            elif status == 'PARSING':
                print("File sedang diurai. Harap tunggu...")
            elif status == 'PARSE_SUCCESS':
                print("Penguraian file berhasil diselesaikan.")
                break
            else:
                print(f"Status file tidak diketahui: {status}. Harap hubungi dukungan teknis.")
                return None
            time.sleep(5)
        # Langkah 7: Mengirim pekerjaan untuk menambahkan dokumen.
        print("Langkah 7: Mengirim pekerjaan untuk menambahkan dokumen.")
        index_add_response = submit_index_add_documents_job(client, workspace_id, index_id, file_id, source_type)
        job_id = index_add_response.body.data.id
        # Langkah 8: Menunggu pekerjaan pengindeksan selesai.
        print("Langkah 8: Menunggu pekerjaan pengindeksan selesai.")
        while True:
            get_index_job_status_response = get_index_job_status(client, workspace_id, job_id, index_id)
            status = get_index_job_status_response.body.data.status
            print(f"Status pekerjaan indeks saat ini: {status}")
            if status == 'COMPLETED':
                break
            time.sleep(5)
        # Langkah 9: Menghapus file lama.
        print("Langkah 9: Menghapus file lama.")
        delete_index_document(client, workspace_id, index_id, old_file_id)
        print("Basis pengetahuan Model Studio berhasil diperbarui.")
        return index_id
    except Exception as e:
        print(f"Terjadi kesalahan: {e}")
        return None

def main():
    if not check_environment_variables():
        print("Pemeriksaan variabel lingkungan gagal.")
        return
    file_path = input("Masukkan jalur lokal file yang diperbarui (misalnya, pada sistem Linux: /path/to/Model_Studio_Phone_Series_Introduction.docx):")
    index_id = input("Masukkan ID basis pengetahuan yang akan diperbarui: ")  # Ini adalah Data.Id yang dikembalikan oleh operasi CreateIndex. Anda juga dapat memperolehnya dari halaman basis pengetahuan di konsol Model Studio.
    old_file_id = input("Masukkan FileId dari file yang akan diganti: ")  # Ini adalah FileId yang dikembalikan oleh operasi AddFile. Anda juga dapat memperolehnya dengan mengklik ikon ID di sebelah nama file pada halaman data aplikasi di konsol Model Studio.
    workspace_id = os.environ.get('WORKSPACE_ID')
    update_knowledge_base(file_path, workspace_id, index_id, old_file_id)

if __name__ == '__main__':
    main()

# Kode contoh ini hanya untuk referensi. Jangan gunakan dalam lingkungan produksi.
import os

from alibabacloud_bailian20231229 import models as bailian_20231229_models
from alibabacloud_bailian20231229.client import Client as bailian20231229Client
from alibabacloud_tea_openapi import models as open_api_models
from alibabacloud_tea_util import models as util_models
from alibabacloud_tea_util.client import Client as UtilClient

def check_environment_variables():
    """Memverifikasi bahwa variabel lingkungan yang diperlukan telah diatur."""
    required_vars = {
        'ALIBABA_CLOUD_ACCESS_KEY_ID': 'Alibaba Cloud AccessKey ID',
        'ALIBABA_CLOUD_ACCESS_KEY_SECRET': 'Alibaba Cloud AccessKey secret',
        'WORKSPACE_ID': 'Alibaba Cloud Model Studio workspace ID'
    }
    missing_vars = []
    for var, description in required_vars.items():
        if not os.environ.get(var):
            missing_vars.append(var)
            print(f"Error: Silakan atur variabel lingkungan {var} ({description})")

    return len(missing_vars) == 0

// Buat klien.
def create_client() -> bailian20231229Client:
    """
    Membuat dan mengonfigurasi klien.

    Returns:
        bailian20231229Client: Klien yang telah dikonfigurasi.
    """
    config = open_api_models.Config(
        access_key_id=os.environ.get('ALIBABA_CLOUD_ACCESS_KEY_ID'),
        access_key_secret=os.environ.get('ALIBABA_CLOUD_ACCESS_KEY_SECRET')
    )
    // Titik akhir berikut untuk cloud publik di wilayah Singapura. Ubah sesuai kebutuhan.
    config.endpoint = 'bailian.ap-southeast-1.aliyuncs.com'
    return bailian20231229Client(config)

// Daftar basis pengetahuan.
def list_indices(client, workspace_id):
    """
    Menampilkan daftar basis pengetahuan dalam ruang kerja tertentu.

    Args:
        client (bailian20231229Client): Klien.
        workspace_id (str): ID ruang kerja.

    Returns:
        Respons dari layanan Alibaba Cloud Model Studio.
    """
    headers = {}
    list_indices_request = bailian_20231229_models.ListIndicesRequest()
    runtime = util_models.RuntimeOptions()
    return client.list_indices_with_options(workspace_id, list_indices_request, headers, runtime)

// Hapus basis pengetahuan.
def delete_index(client, workspace_id, index_id):
    """
    Menghapus secara permanen basis pengetahuan yang ditentukan.

    Args:
        client (bailian20231229Client): Klien.
        workspace_id (str): ID ruang kerja.
        index_id (str): ID basis pengetahuan.

    Returns:
        Respons dari layanan Alibaba Cloud Model Studio.
    """
    headers = {}
    delete_index_request = bailian_20231229_models.DeleteIndexRequest(
        index_id=index_id
    )
    runtime = util_models.RuntimeOptions()
    return client.delete_index_with_options(workspace_id, delete_index_request, headers, runtime)

def main():
    if not check_environment_variables():
        print("Pemeriksaan variabel lingkungan gagal.")
        return
    try:
        start_option = input(
            "Pilih operasi yang akan dilakukan:\n1. Daftar basis pengetahuan\n2. Hapus basis pengetahuan\nMasukkan opsi (1 atau 2):")
        if start_option == '1':
            // Daftar basis pengetahuan.
            print("\nMenampilkan daftar basis pengetahuan...")
            workspace_id = os.environ.get('WORKSPACE_ID')
            client = create_client()
            list_indices_response = list_indices(client, workspace_id)
            print(UtilClient.to_jsonstring(list_indices_response.body.data))
        elif start_option == '2':
            print("\nMenghapus basis pengetahuan...")
            workspace_id = os.environ.get('WORKSPACE_ID')
            index_id = input("Masukkan ID basis pengetahuan: ")  // Ini adalah nilai `Data.Id` dari respons `CreateIndex`. Anda juga dapat memperolehnya dari halaman basis pengetahuan di konsol Model Studio.
            // Konfirmasi sebelum penghapusan.
            while True:
                confirm = input(f"Apakah Anda yakin ingin menghapus secara permanen basis pengetahuan {index_id}? (y/n): ").strip().lower()
                if confirm == 'y':
                    break
                elif confirm == 'n':
                    print("Penghapusan dibatalkan.")
                    return
                else:
                    print("Input tidak valid. Harap masukkan y atau n.")
            client = create_client()
            resp = delete_index(client, workspace_id, index_id)
            if resp.body.status == 200:
                print(f"Basis pengetahuan {index_id} berhasil dihapus.")
            else:
                err_info = UtilClient.to_jsonstring(resp.body)
                print(f"Terjadi kesalahan: {err_info}")
        else:
            print("Opsi tidak valid. Keluar dari program.")
            return
    except Exception as e:
        print(f"Terjadi kesalahan: {e}")
        return

if __name__ == '__main__':
    main()