kyo-takano

kyo-takano

import gzip

def gzip_search(query: str, candidate_chunks: list[str], top_k: int=1):
    """
    文字列ベースで類似したテキストチャンクを推定するアルゴリズム.
    `query`, `chunk`, および`query + " " + chunk`をそれぞれgzipで圧縮し、編集距離のようなものをベースに評価する.

    Parameters:
        query (str): 検索クエリとして使用する文字列.
        top_k (int, optional): 返される類似チャンクの上位k個を指定する (default: 1).

kyo-takano

kyo-takano

kyo-takano

""" Iteratively generate, review, and improve a code snippet using OpenAI's `Completion` and `Embedding` APIs """

import os
import openai
import textwrap
from scipy.spatial import distance

import matplotlib.pyplot as plt
from tqdm import trange
from IPython.display import display, Code, clear_output