formatting
This commit is contained in:
@@ -26,11 +26,21 @@ from pipecat.frames.frames import (
|
||||
|
||||
class BaseOpenPipeLLMService(LLMService):
|
||||
|
||||
def __init__(self, model: str, c_id=None, api_key=None, openpipe_api_key=None, openpipe_base_url=None, prompt=None):
|
||||
def __init__(
|
||||
self,
|
||||
model: str,
|
||||
c_id=None,
|
||||
api_key=None,
|
||||
openpipe_api_key=None,
|
||||
openpipe_base_url=None,
|
||||
prompt=None):
|
||||
super().__init__()
|
||||
self._model = model
|
||||
self._client = self.create_client(api_key=api_key, openpipe_api_key=openpipe_api_key, openpipe_base_url=openpipe_base_url)
|
||||
self.c_id=c_id if c_id else secrets.token_urlsafe(16)
|
||||
self._client = self.create_client(
|
||||
api_key=api_key,
|
||||
openpipe_api_key=openpipe_api_key,
|
||||
openpipe_base_url=openpipe_base_url)
|
||||
self.c_id = c_id if c_id else secrets.token_urlsafe(16)
|
||||
self.prompt = prompt
|
||||
logger.debug(f"Client Created: {self._client}")
|
||||
|
||||
@@ -71,7 +81,7 @@ class BaseOpenPipeLLMService(LLMService):
|
||||
messages=messages,
|
||||
openpipe={
|
||||
"tags": {"conversation_id": self.c_id,
|
||||
"prompt":self.prompt},
|
||||
"prompt": self.prompt},
|
||||
"log_request": True
|
||||
}
|
||||
)
|
||||
@@ -142,6 +152,7 @@ class BaseOpenPipeLLMService(LLMService):
|
||||
if context:
|
||||
await self._process_context(context)
|
||||
|
||||
|
||||
class OpenPipeLLMService(BaseOpenPipeLLMService):
|
||||
|
||||
def __init__(self, model="gpt-4o", cli_id=None, **kwargs):
|
||||
|
||||
Reference in New Issue
Block a user