feat: OpenRouter engine, with OPENROUTER_API_KEY env var
This commit is contained in:
1 parent
4e15b44d39
commit
50cd677515
7 files changed
+58
-119
No files matched your search
+8
-1
@@ -10,6 +10,7 @@ from tortoise.expressions import Q
|
||||
from quart import Quart, Response, request, send_from_directory, jsonify, websocket
|
||||
from qdrant_client import QdrantClient
|
||||
import replicate
|
||||
from openrouter import OpenRouter
|
||||
|
||||
import database as db
|
||||
import agent as agent_mod
|
||||
@@ -36,7 +37,13 @@ class App(ChatMixin):
|
||||
if replicate_api_token:
|
||||
self.replicate_client = replicate.Client(api_token=replicate_api_token)
|
||||
else:
|
||||
raise Exception("Horde only works with a replicate connection. Please provide an API Token.")
|
||||
self.replicate_client = None
|
||||
|
||||
openrouter_api_key = os.getenv("OPENROUTER_API_KEY")
|
||||
if openrouter_api_key:
|
||||
self.openrouter_client = OpenRouter(api_key=openrouter_api_key)
|
||||
else:
|
||||
self.openrouter_client = None
|
||||
|
||||
self.agent_classes = [
|
||||
agents_mod.ExpertAgent,
|
||||
|
||||
Reference in new issue
Block a user