llm/tests/test_embed.py

import llm


def test_demo_plugin():
    model = llm.get_embedding_model("embed-demo")
    assert model.embed("hello world") == [5, 5] + [0] * 14
Initial CLI support and plugin hook for embeddings, refs #185 * Embeddings plugin hook + OpenAI implementation * llm.get_embedding_model(name) function * llm embed command, for returning embeddings or saving them to SQLite * Tests using an EmbedDemo embedding model * llm embed-models list and emeb-models default commands * llm embed-db path and llm embed-db collections commands 2023-08-28 05:24:10 +00:00			`import llm`


			`def test_demo_plugin():`
			`model = llm.get_embedding_model("embed-demo")`
			`assert model.embed("hello world") == [5, 5] + [0] * 14`