Skip to content
Snippets Groups Projects
Commit 314005d2 authored by Ismail Ashraq's avatar Ismail Ashraq
Browse files

add LlamaCppLLM to init

parent 5e6fd2f0
No related branches found
No related tags found
No related merge requests found
from semantic_router.llms.base import BaseLLM from semantic_router.llms.base import BaseLLM
from semantic_router.llms.cohere import CohereLLM from semantic_router.llms.cohere import CohereLLM
from semantic_router.llms.llamacpp import LlamaCppLLM
from semantic_router.llms.mistral import MistralAILLM from semantic_router.llms.mistral import MistralAILLM
from semantic_router.llms.openai import OpenAILLM from semantic_router.llms.openai import OpenAILLM
from semantic_router.llms.openrouter import OpenRouterLLM from semantic_router.llms.openrouter import OpenRouterLLM
...@@ -8,6 +9,7 @@ from semantic_router.llms.zure import AzureOpenAILLM ...@@ -8,6 +9,7 @@ from semantic_router.llms.zure import AzureOpenAILLM
__all__ = [ __all__ = [
"BaseLLM", "BaseLLM",
"OpenAILLM", "OpenAILLM",
"LlamaCppLLM",
"OpenRouterLLM", "OpenRouterLLM",
"CohereLLM", "CohereLLM",
"AzureOpenAILLM", "AzureOpenAILLM",
......
...@@ -2,8 +2,6 @@ from contextlib import contextmanager ...@@ -2,8 +2,6 @@ from contextlib import contextmanager
from pathlib import Path from pathlib import Path
from typing import Any, Optional from typing import Any, Optional
# from llama_cpp import Llama, LlamaGrammar
from semantic_router.llms.base import BaseLLM from semantic_router.llms.base import BaseLLM
from semantic_router.schema import Message from semantic_router.schema import Message
from semantic_router.utils.logger import logger from semantic_router.utils.logger import logger
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment