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