INNER CODE UNIT · Python

embedding

tensorchord/modelz-llm · client.py:25

def embedding():
    emb = openai.Embedding.create(
        input=["Once upon a time", "There was a frog", "Who lived in a well"],
        model="text-embedding-ada-002",
    )
    print(emb)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("-c", "--chat", action="store_true")
    parser.add_argument("-e", "--embedding", action="store_true")
    args = parser.parse_args()

    if args.chat:
        chat()
    if args.embedding:
        embedding()

View source record →