是的，OpenAI可以为任何输入文本创建嵌入--即使是代码。你只需要在它的get_embedding()函数调用中传递正确的引擎或模型。

# Third-party imports
import openai

from openai.embeddings_utils import get_embedding

openai.api_key = OPENAI_SEC_KEY

embedding = get_embedding("""
    def sample_code():
        print("Hello from IamAshKS !!!")
""", engine="code-search-babbage-code-001")

print()
print(f"{embedding=}")
print(f"{len(embedding)=}")

# OUTPUT:
# embedding=[-0.007094269152730703, 0.006055716425180435, -0.005044757854193449, ...]
# len(embedding)=2048

embedding = get_embedding("""
import Random
a = random.randint(1,12)
b = random.randint(1,12)
for i in range(10):
    question = "What is "+a+" x "+b+"? "
    answer = input(question)
    if answer = a*b
        print (Well done!)
    else:
        print("No.")
""", engine="code-search-babbage-code-001")

print()
print(f"{embedding=}")
print(f"{len(embedding)=}")

# OUTPUT:
# embedding=[-0.011341490782797337, -0.005919027142226696, 0.0011923711281269789, ...]
# len(embedding)=2048

• • 注：**您可以使用get_embedding()的engine参数更换型号或发动机。

上面给出的代码可以让你嵌入任何代码。还有一个名为code-search-ada-code-001的代码搜索引擎/模型，但它不如code-search-babbage-code-001强大，我在回答这个问题时使用了code-search-babbage-code-001。如果你也想做代码搜索，请阅读下面的参考文献。

• • 参考资料：**
• https://openai.com/blog/introducing-text-and-code-embeddings/
• https://beta.openai.com/docs/guides/embeddings/code-search-embeddings
• https://beta.openai.com/docs/guides/embeddings/use-cases#expander-7

函数get_embedding将为我们提供一个输入文本的嵌入。
OpenAI的规范代码如下：https://github.com/openai/openai-python/blob/main/examples/embeddings/Get_embeddings.ipynb

import openai
from tenacity import retry, wait_random_exponential, stop_after_attempt

@retry(wait=wait_random_exponential(min=1, max=20), stop=stop_after_attempt(6))
def get_embedding(text: str, engine="text-similarity-davinci-001") -> List[float]:

    # replace newlines, which can negatively affect performance.
    text = text.replace("\n", " ")

    return openai.Embedding.create(input=[text], engine=engine)["data"][0]["embedding"]

embedding = get_embedding("Sample query text goes here", engine="text-search-ada-query-001")
print(len(embedding))