freeplay 0.2.42__py3-none-any.whl → 0.3.0a2__py3-none-any.whl

freeplay/record.py

@@ -1,113 +0,0 @@
-import logging
-from abc import abstractmethod, ABC
-from dataclasses import dataclass
-from typing import Dict, Union
-from typing import Optional
-
-from . import api_support
-from .completions import PromptTemplateWithMetadata, OpenAIFunctionCall
-from .llm_parameters import LLMParameters
-from .model import InputVariables
-
-logger = logging.getLogger(__name__)
-
-
-@dataclass
-class RecordCallFields:
-    completion_content: str
-    completion_is_complete: bool
-    end: float
-    formatted_prompt: str
-    session_id: str
-    start: float
-    target_template: PromptTemplateWithMetadata
-    variables: InputVariables
-    tag: str
-    test_run_id: Optional[str]
-    test_case_id: Optional[str]
-    record_format_type: Optional[str]
-    model: Optional[str]
-    provider: Optional[str]
-    llm_parameters: Optional[LLMParameters]
-    function_call_response: Optional[OpenAIFunctionCall] = None
-    custom_metadata: Optional[Dict[str, Union[str, int, float]]] = None
-
-
-class RecordProcessor(ABC):
-    @abstractmethod
-    def record_call(
-            self,
-            record_call: RecordCallFields,
-    ) -> None:
-        pass
-
-
-class NoOpRecorder(RecordProcessor):
-    def record_call(
-            self,
-            record_call: RecordCallFields,
-    ) -> None:
-        pass
-
-
-no_op_recorder = NoOpRecorder()
-
-
-class DefaultRecordProcessor(RecordProcessor):
-
-    def __init__(
-            self,
-            freeplay_api_key: str,
-            api_base: str
-    ) -> None:
-        self.api_base = api_base
-        self.freeplay_api_key = freeplay_api_key
-
-    def record_call(
-            self,
-            record_call: RecordCallFields
-    ) -> None:
-        record_payload = {
-            "session_id": record_call.session_id,
-            "project_version_id": record_call.target_template.prompt_template_version_id,
-            "prompt_template_id": record_call.target_template.prompt_template_id,
-            "start_time": record_call.start,
-            "end_time": record_call.end,
-            "tag": record_call.tag,
-            "inputs": record_call.variables,
-            "prompt_content": record_call.formatted_prompt,
-            "return_content": record_call.completion_content,
-            "format_type": record_call.record_format_type,
-            "is_complete": record_call.completion_is_complete,
-            "model": record_call.model,
-            "provider": record_call.provider,
-            "llm_parameters": record_call.llm_parameters,
-        }
-
-        if record_call.custom_metadata is not None:
-            record_payload['custom_metadata'] = record_call.custom_metadata
-
-        if record_call.function_call_response is not None:
-            record_payload['function_call_response'] = record_call.function_call_response
-
-        if record_call.test_run_id is not None:
-            record_payload['test_run_id'] = record_call.test_run_id
-
-        if record_call.test_case_id is not None:
-            record_payload['test_case_id'] = record_call.test_case_id
-
-        try:
-            recorded_response = api_support.post_raw(
-                api_key=self.freeplay_api_key,
-                url=f'{self.api_base}/v1/record',
-                payload=record_payload
-            )
-            recorded_response.raise_for_status()
-        except Exception as e:
-            status_code = -1
-            if hasattr(e, 'response') and hasattr(e.response, 'status_code'):
-                status_code = e.response.status_code
-            logger.warning(
-                f'There was an error recording to Freeplay. Call will not be logged. '
-                f'Status: {status_code}. {e.__class__}'
-            )

freeplay/thin/__init__.py

@@ -1,14 +0,0 @@
-from .freeplay_thin import Freeplay
-from .resources.prompts import PromptInfo
-from .resources.recordings import CallInfo, ResponseInfo, RecordPayload, TestRunInfo
-from .resources.sessions import SessionInfo
-
-__all__ = [
-    'Freeplay',
-    'CallInfo',
-    'PromptInfo',
-    'RecordPayload',
-    'ResponseInfo',
-    'SessionInfo',
-    'TestRunInfo',
-]

freeplay/thin/freeplay_thin.py

@@ -1,42 +0,0 @@
-from typing import Optional
-
-from freeplay.errors import FreeplayConfigurationError
-from freeplay.record import DefaultRecordProcessor
-from freeplay.support import CallSupport
-from freeplay.thin.resources.customer_feedback import CustomerFeedback
-from freeplay.thin.resources.prompts import Prompts, APITemplateResolver, TemplateResolver
-from freeplay.thin.resources.recordings import Recordings
-from freeplay.thin.resources.sessions import Sessions
-from freeplay.thin.resources.test_runs import TestRuns
-
-
-class Freeplay:
-    def __init__(
-            self,
-            freeplay_api_key: str,
-            api_base: str,
-            template_resolver: Optional[TemplateResolver] = None
-    ) -> None:
-        if not freeplay_api_key or not freeplay_api_key.strip():
-            raise FreeplayConfigurationError("Freeplay API key not set. It must be set to the Freeplay API.")
-
-        self.call_support = CallSupport(
-            freeplay_api_key,
-            api_base,
-            DefaultRecordProcessor(freeplay_api_key, api_base)
-        )
-        self.freeplay_api_key = freeplay_api_key
-        self.api_base = api_base
-
-        resolver: TemplateResolver
-        if template_resolver is None:
-            resolver = APITemplateResolver(self.call_support)
-        else:
-            resolver = template_resolver
-
-        # Resources ========
-        self.customer_feedback = CustomerFeedback(self.call_support)
-        self.prompts = Prompts(self.call_support, resolver)
-        self.recordings = Recordings(self.call_support)
-        self.sessions = Sessions()
-        self.test_runs = TestRuns(self.call_support)

freeplay-0.2.42.dist-info/RECORD

@@ -1,27 +0,0 @@
-freeplay/__init__.py,sha256=74A9S9hmLq9BNHsdx0-37yDxlSukudNl9bJ0TE60Z30,61
-freeplay/api_support.py,sha256=Tmc6VRcmPcIgXrDNJXz0VbBD969t5QHAyV3yZlMsYRI,2331
-freeplay/completions.py,sha256=F1rMBtQaCtn0rBQqvCurkV25g8gLtwnEod5rRvf-txY,1176
-freeplay/errors.py,sha256=bPqsw32YX-xSr7O-G49M0sSFF7mq-YF1WGq928UV47s,631
-freeplay/flavors.py,sha256=9jVvhym9L5zyWM3RTBViibIgG6f6Fh1_63uMEVZAFT4,17259
-freeplay/freeplay.py,sha256=BLlr4YXh3a624xcM93KvZGsiIGSMrIMFM07lLyVCIao,17089
-freeplay/freeplay_cli.py,sha256=k9QPBedcFqpy9Ec9k7DT_y-NpVxHqRF5jKPTaQBBs1Q,3863
-freeplay/llm_parameters.py,sha256=bQbfuC8EICF0XMZQa5pwI3FkQqxmCUVqHO3gYHy3Tg8,898
-freeplay/model.py,sha256=7BRAuyxsLl8X-ue8wXPmAiKYp4OaYI-HkaGc3t5qtMU,534
-freeplay/provider_config.py,sha256=hruf3Khusrwb76_-hv7ouuxmvJuaRyC1UxIw7XlJx8A,1416
-freeplay/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-freeplay/record.py,sha256=69vO9vYGFL8gLrvRqLzlvodVNo05GOzKDJnCyhUha08,3615
-freeplay/support.py,sha256=ERXFKxZHl-TfgjJJYW-ssAFdSz_CJIs25wFlCPp4DWc,14535
-freeplay/thin/__init__.py,sha256=XgU_eMmTpOPCa9w5mVOyxPgwAgRFfVdRWDazYGYZtQ0,351
-freeplay/thin/freeplay_thin.py,sha256=0w_ECjassczsHJUcpzPxEHUDbiGo7Ey7Ba7vIaWc5ds,1619
-freeplay/thin/resources/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-freeplay/thin/resources/customer_feedback.py,sha256=aTM7Eez7iYmjXSpRqkHxf4pi6xBrzVnMiQCEJVfPGvg,527
-freeplay/thin/resources/prompts.py,sha256=3sfH1k1AAXRfgHxfV6Acws0fAiMswgE93auRKWLGVIQ,10135
-freeplay/thin/resources/recordings.py,sha256=H_l5_J7nhoL5p0P37JsEC1_XROKc0PDsrV1UomNhy3E,5670
-freeplay/thin/resources/sessions.py,sha256=ioWdeTM9BSrEWKrFH66ysQIw5kCTlCVopJDmWtFgHz0,868
-freeplay/thin/resources/test_runs.py,sha256=mhRFjR-7z_1wGAkwRjDjALBjT1E6pen91DF8kZHjbLg,1457
-freeplay/utils.py,sha256=cRCCIzVqWNDKlTI-DDhXGyCkplbd-X4qzDs__aUpvww,1840
-freeplay-0.2.42.dist-info/LICENSE,sha256=_jzIw45hB1XHGxiQ8leZ0GH_X7bR_a8qgxaqnHbCUOo,1064
-freeplay-0.2.42.dist-info/METADATA,sha256=xDUBWu3ZrV1J0cNiMubC_0PdMD0ZJDeBwq3urjPejSc,1678
-freeplay-0.2.42.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-freeplay-0.2.42.dist-info/entry_points.txt,sha256=32s3rf2UUCqiJT4jnClEXZhdXlvl30uwpcxz-Gsy4UU,54
-freeplay-0.2.42.dist-info/RECORD,,