edsl 0.1.47__py3-none-any.whl → 0.1.48__py3-none-any.whl

edsl/jobs/JobsPrompts.py

@@ -1,354 +0,0 @@
-import time
-import logging
-from typing import List, TYPE_CHECKING
-
-from edsl.results.Dataset import Dataset
-
-if TYPE_CHECKING:
-    from edsl.jobs import Jobs
-
-    # from edsl.jobs.interviews.Interview import Interview
-    # from edsl.results.Dataset import Dataset
-    # from edsl.agents.AgentList import AgentList
-    # from edsl.scenarios.ScenarioList import ScenarioList
-    # from edsl.surveys.Survey import Survey
-
-from edsl.jobs.FetchInvigilator import FetchInvigilator
-from edsl.data.CacheEntry import CacheEntry
-
-logger = logging.getLogger(__name__)
-
-
-class JobsPrompts:
-    def __init__(self, jobs: "Jobs"):
-        self.interviews = jobs.interviews()
-        self.agents = jobs.agents
-        self.scenarios = jobs.scenarios
-        self.survey = jobs.survey
-        self._price_lookup = None
-        self._agent_lookup = {agent: idx for idx, agent in enumerate(self.agents)}
-        self._scenario_lookup = {
-            scenario: idx for idx, scenario in enumerate(self.scenarios)
-        }
-
-    @property
-    def price_lookup(self):
-        if self._price_lookup is None:
-            from edsl.coop.coop import Coop
-
-            c = Coop()
-            self._price_lookup = c.fetch_prices()
-        return self._price_lookup
-
-    def prompts(self, iterations=1) -> "Dataset":
-        """Return a Dataset of prompts that will be used.
-
-        >>> from edsl.jobs import Jobs
-        >>> Jobs.example().prompts()
-        Dataset(...)
-        """
-        interviews = self.interviews
-        interview_indices = []
-        question_names = []
-        user_prompts = []
-        system_prompts = []
-        scenario_indices = []
-        agent_indices = []
-        models = []
-        costs = []
-        cache_keys = []
-
-        for interview_index, interview in enumerate(interviews):
-            logger.info(f"Processing interview {interview_index} of {len(interviews)}")
-            interview_start = time.time()
-
-            # Fetch invigilators timing
-            invig_start = time.time()
-            invigilators = [
-                FetchInvigilator(interview)(question)
-                for question in interview.survey.questions
-            ]
-            invig_end = time.time()
-            logger.debug(
-                f"Time taken to fetch invigilators: {invig_end - invig_start:.4f}s"
-            )
-
-            # Process prompts timing
-            prompts_start = time.time()
-            for _, invigilator in enumerate(invigilators):
-                # Get prompts timing
-                get_prompts_start = time.time()
-                prompts = invigilator.get_prompts()
-                get_prompts_end = time.time()
-                logger.debug(
-                    f"Time taken to get prompts: {get_prompts_end - get_prompts_start:.4f}s"
-                )
-
-                user_prompt = prompts["user_prompt"]
-                system_prompt = prompts["system_prompt"]
-                user_prompts.append(user_prompt)
-                system_prompts.append(system_prompt)
-
-                # Index lookups timing
-                index_start = time.time()
-                agent_index = self._agent_lookup[invigilator.agent]
-                agent_indices.append(agent_index)
-                interview_indices.append(interview_index)
-                scenario_index = self._scenario_lookup[invigilator.scenario]
-                scenario_indices.append(scenario_index)
-                index_end = time.time()
-                logger.debug(
-                    f"Time taken for index lookups: {index_end - index_start:.4f}s"
-                )
-
-                # Model and question name assignment timing
-                assign_start = time.time()
-                models.append(invigilator.model.model)
-                question_names.append(invigilator.question.question_name)
-                assign_end = time.time()
-                logger.debug(
-                    f"Time taken for assignments: {assign_end - assign_start:.4f}s"
-                )
-
-                # Cost estimation timing
-                cost_start = time.time()
-                prompt_cost = self.estimate_prompt_cost(
-                    system_prompt=system_prompt,
-                    user_prompt=user_prompt,
-                    price_lookup=self.price_lookup,
-                    inference_service=invigilator.model._inference_service_,
-                    model=invigilator.model.model,
-                )
-                cost_end = time.time()
-                logger.debug(
-                    f"Time taken to estimate prompt cost: {cost_end - cost_start:.4f}s"
-                )
-                costs.append(prompt_cost["cost_usd"])
-
-                # Cache key generation timing
-                cache_key_gen_start = time.time()
-                for iteration in range(iterations):
-                    cache_key = CacheEntry.gen_key(
-                        model=invigilator.model.model,
-                        parameters=invigilator.model.parameters,
-                        system_prompt=system_prompt,
-                        user_prompt=user_prompt,
-                        iteration=iteration,
-                    )
-                    cache_keys.append(cache_key)
-
-                cache_key_gen_end = time.time()
-                logger.debug(
-                    f"Time taken to generate cache key: {cache_key_gen_end - cache_key_gen_start:.4f}s"
-                )
-                logger.debug("-" * 50)  # Separator between iterations
-
-            prompts_end = time.time()
-            logger.info(
-                f"Time taken to process prompts: {prompts_end - prompts_start:.4f}s"
-            )
-
-            interview_end = time.time()
-            logger.info(
-                f"Overall time taken for interview: {interview_end - interview_start:.4f}s"
-            )
-            logger.info("Time breakdown:")
-            logger.info(f"  Invigilators: {invig_end - invig_start:.4f}s")
-            logger.info(f"  Prompts processing: {prompts_end - prompts_start:.4f}s")
-            logger.info(
-                f"  Other overhead: {(interview_end - interview_start) - ((invig_end - invig_start) + (prompts_end - prompts_start)):.4f}s"
-            )
-
-        d = Dataset(
-            [
-                {"user_prompt": user_prompts},
-                {"system_prompt": system_prompts},
-                {"interview_index": interview_indices},
-                {"question_name": question_names},
-                {"scenario_index": scenario_indices},
-                {"agent_index": agent_indices},
-                {"model": models},
-                {"estimated_cost": costs},
-                {"cache_key": cache_keys},
-            ]
-        )
-        return d
-
-    @staticmethod
-    def estimate_prompt_cost(
-        system_prompt: str,
-        user_prompt: str,
-        price_lookup: dict,
-        inference_service: str,
-        model: str,
-    ) -> dict:
-        """Estimates the cost of a prompt. Takes piping into account."""
-        import math
-
-        def get_piping_multiplier(prompt: str):
-            """Returns 2 if a prompt includes Jinja braces, and 1 otherwise."""
-
-            if "{{" in prompt and "}}" in prompt:
-                return 2
-            return 1
-
-        # Look up prices per token
-        key = (inference_service, model)
-
-        try:
-            relevant_prices = price_lookup[key]
-
-            service_input_token_price = float(
-                relevant_prices["input"]["service_stated_token_price"]
-            )
-            service_input_token_qty = float(
-                relevant_prices["input"]["service_stated_token_qty"]
-            )
-            input_price_per_token = service_input_token_price / service_input_token_qty
-
-            service_output_token_price = float(
-                relevant_prices["output"]["service_stated_token_price"]
-            )
-            service_output_token_qty = float(
-                relevant_prices["output"]["service_stated_token_qty"]
-            )
-            output_price_per_token = (
-                service_output_token_price / service_output_token_qty
-            )
-
-        except KeyError:
-            # A KeyError is likely to occur if we cannot retrieve prices (the price_lookup dict is empty)
-            # Use a sensible default
-
-            import warnings
-
-            warnings.warn(
-                "Price data could not be retrieved. Using default estimates for input and output token prices. Input: $1.00 / 1M tokens; Output: $1.00 / 1M tokens"
-            )
-            input_price_per_token = 0.000001  # $1.00 / 1M tokens
-            output_price_per_token = 0.000001  # $1.00 / 1M tokens
-
-        # Compute the number of characters (double if the question involves piping)
-        user_prompt_chars = len(str(user_prompt)) * get_piping_multiplier(
-            str(user_prompt)
-        )
-        system_prompt_chars = len(str(system_prompt)) * get_piping_multiplier(
-            str(system_prompt)
-        )
-
-        # Convert into tokens (1 token approx. equals 4 characters)
-        input_tokens = (user_prompt_chars + system_prompt_chars) // 4
-
-        output_tokens = math.ceil(0.75 * input_tokens)
-
-        cost = (
-            input_tokens * input_price_per_token
-            + output_tokens * output_price_per_token
-        )
-
-        return {
-            "input_tokens": input_tokens,
-            "output_tokens": output_tokens,
-            "cost_usd": cost,
-        }
-
-    def estimate_job_cost_from_external_prices(
-        self, price_lookup: dict, iterations: int = 1
-    ) -> dict:
-        """
-        Estimates the cost of a job according to the following assumptions:
-
-        - 1 token = 4 characters.
-        - For each prompt, output tokens = input tokens * 0.75, rounded up to the nearest integer.
-
-        price_lookup is an external pricing dictionary.
-        """
-
-        import pandas as pd
-
-        interviews = self.interviews
-        data = []
-        for interview in interviews:
-            invigilators = [
-                FetchInvigilator(interview)(question)
-                for question in self.survey.questions
-            ]
-            for invigilator in invigilators:
-                prompts = invigilator.get_prompts()
-
-                # By this point, agent and scenario data has already been added to the prompts
-                user_prompt = prompts["user_prompt"]
-                system_prompt = prompts["system_prompt"]
-                inference_service = invigilator.model._inference_service_
-                model = invigilator.model.model
-
-                prompt_cost = self.estimate_prompt_cost(
-                    system_prompt=system_prompt,
-                    user_prompt=user_prompt,
-                    price_lookup=price_lookup,
-                    inference_service=inference_service,
-                    model=model,
-                )
-
-                data.append(
-                    {
-                        "user_prompt": user_prompt,
-                        "system_prompt": system_prompt,
-                        "estimated_input_tokens": prompt_cost["input_tokens"],
-                        "estimated_output_tokens": prompt_cost["output_tokens"],
-                        "estimated_cost_usd": prompt_cost["cost_usd"],
-                        "inference_service": inference_service,
-                        "model": model,
-                    }
-                )
-
-        df = pd.DataFrame.from_records(data)
-
-        df = (
-            df.groupby(["inference_service", "model"])
-            .agg(
-                {
-                    "estimated_cost_usd": "sum",
-                    "estimated_input_tokens": "sum",
-                    "estimated_output_tokens": "sum",
-                }
-            )
-            .reset_index()
-        )
-        df["estimated_cost_usd"] = df["estimated_cost_usd"] * iterations
-        df["estimated_input_tokens"] = df["estimated_input_tokens"] * iterations
-        df["estimated_output_tokens"] = df["estimated_output_tokens"] * iterations
-
-        estimated_costs_by_model = df.to_dict("records")
-
-        estimated_total_cost = sum(
-            model["estimated_cost_usd"] for model in estimated_costs_by_model
-        )
-        estimated_total_input_tokens = sum(
-            model["estimated_input_tokens"] for model in estimated_costs_by_model
-        )
-        estimated_total_output_tokens = sum(
-            model["estimated_output_tokens"] for model in estimated_costs_by_model
-        )
-
-        output = {
-            "estimated_total_cost_usd": estimated_total_cost,
-            "estimated_total_input_tokens": estimated_total_input_tokens,
-            "estimated_total_output_tokens": estimated_total_output_tokens,
-            "model_costs": estimated_costs_by_model,
-        }
-
-        return output
-
-    def estimate_job_cost(self, iterations: int = 1) -> dict:
-        """
-        Estimates the cost of a job according to the following assumptions:
-
-        - 1 token = 4 characters.
-        - For each prompt, output tokens = input tokens * 0.75, rounded up to the nearest integer.
-
-        Fetches prices from Coop.
-        """
-        return self.estimate_job_cost_from_external_prices(
-            price_lookup=self.price_lookup, iterations=iterations
-        )

edsl/jobs/buckets/BucketCollection.py

@@ -1,134 +0,0 @@
-from typing import Optional
-from collections import UserDict
-from edsl.jobs.buckets.TokenBucket import TokenBucket
-from edsl.jobs.buckets.ModelBuckets import ModelBuckets
-
-# from functools import wraps
-from threading import RLock
-
-from edsl.jobs.decorators import synchronized_class
-
-
-@synchronized_class
-class BucketCollection(UserDict):
-    """A Jobs object will have a whole collection of model buckets, as multiple models could be used.
-
-    The keys here are the models, and the values are the ModelBuckets objects.
-    Models themselves are hashable, so this works.
-    """
-
-    def __init__(self, infinity_buckets: bool = False):
-        """Create a new BucketCollection.
-        An infinity bucket is a bucket that never runs out of tokens or requests.
-        """
-        super().__init__()
-        self.infinity_buckets = infinity_buckets
-        self.models_to_services = {}
-        self.services_to_buckets = {}
-        self._lock = RLock()
-
-        from edsl.config import CONFIG
-        import os
-
-        url = os.environ.get("EDSL_REMOTE_TOKEN_BUCKET_URL", None)
-
-        if url == "None" or url is None:
-            self.remote_url = None
-            # print(f"Using remote token bucket URL: {url}")
-        else:
-            self.remote_url = url
-
-    @classmethod
-    def from_models(
-        cls, models_list: list, infinity_buckets: bool = False
-    ) -> "BucketCollection":
-        """Create a BucketCollection from a list of models."""
-        bucket_collection = cls(infinity_buckets=infinity_buckets)
-        for model in models_list:
-            bucket_collection.add_model(model)
-        return bucket_collection
-
-    def get_tokens(
-        self, model: "LanguageModel", bucket_type: str, num_tokens: int
-    ) -> int:
-        """Get the number of tokens remaining in the bucket."""
-        relevant_bucket = getattr(self[model], bucket_type)
-        return relevant_bucket.get_tokens(num_tokens)
-
-    def __repr__(self):
-        return f"BucketCollection({self.data})"
-
-    def add_model(self, model: "LanguageModel") -> None:
-        """Adds a model to the bucket collection.
-
-        This will create the token and request buckets for the model."""
-
-        # compute the TPS and RPS from the model
-        if not self.infinity_buckets:
-            TPS = model.tpm / 60.0
-            RPS = model.rpm / 60.0
-        else:
-            TPS = float("inf")
-            RPS = float("inf")
-
-        if model.model not in self.models_to_services:
-            service = model._inference_service_
-            if service not in self.services_to_buckets:
-                requests_bucket = TokenBucket(
-                    bucket_name=service,
-                    bucket_type="requests",
-                    capacity=RPS,
-                    refill_rate=RPS,
-                    remote_url=self.remote_url,
-                )
-                tokens_bucket = TokenBucket(
-                    bucket_name=service,
-                    bucket_type="tokens",
-                    capacity=TPS,
-                    refill_rate=TPS,
-                    remote_url=self.remote_url,
-                )
-                self.services_to_buckets[service] = ModelBuckets(
-                    requests_bucket, tokens_bucket
-                )
-            self.models_to_services[model.model] = service
-            self[model] = self.services_to_buckets[service]
-        else:
-            self[model] = self.services_to_buckets[self.models_to_services[model.model]]
-
-    def update_from_key_lookup(self, key_lookup: "KeyLookup") -> None:
-        """Updates the bucket collection rates based on model RPM/TPM from KeyLookup"""
-
-        for model_name, service in self.models_to_services.items():
-            if service in key_lookup and not self.infinity_buckets:
-
-                if key_lookup[service].rpm is not None:
-                    new_rps = key_lookup[service].rpm / 60.0
-                    new_requests_bucket = TokenBucket(
-                        bucket_name=service,
-                        bucket_type="requests",
-                        capacity=new_rps,
-                        refill_rate=new_rps,
-                        remote_url=self.remote_url,
-                    )
-                    self.services_to_buckets[service].requests_bucket = (
-                        new_requests_bucket
-                    )
-
-                if key_lookup[service].tpm is not None:
-                    new_tps = key_lookup[service].tpm / 60.0
-                    new_tokens_bucket = TokenBucket(
-                        bucket_name=service,
-                        bucket_type="tokens",
-                        capacity=new_tps,
-                        refill_rate=new_tps,
-                        remote_url=self.remote_url,
-                    )
-                    self.services_to_buckets[service].tokens_bucket = new_tokens_bucket
-
-    def visualize(self) -> dict:
-        """Visualize the token and request buckets for each model."""
-        plots = {}
-        for model in self:
-            plots[model] = self[model].visualize()
-        return plots

edsl/jobs/buckets/ModelBuckets.py

@@ -1,65 +0,0 @@
-# from edsl.jobs.buckets.TokenBucket import TokenBucket
-
-
-class ModelBuckets:
-    """A class to represent the token and request buckets for a model.
-
-    Most LLM model services have limits both on requests-per-minute (RPM) and tokens-per-minute (TPM).
-    A request is one call to the service. The number of tokens required for a request depends on parameters.
-    """
-
-    def __init__(self, requests_bucket: "TokenBucket", tokens_bucket: "TokenBucket"):
-        """Initialize the model buckets.
-
-        The requests bucket captures requests per unit of time.
-        The tokens bucket captures the number of language model tokens.
-
-        """
-        self.requests_bucket = requests_bucket
-        self.tokens_bucket = tokens_bucket
-
-    def __add__(self, other: "ModelBuckets"):
-        """Combine two model buckets."""
-        return ModelBuckets(
-            requests_bucket=self.requests_bucket + other.requests_bucket,
-            tokens_bucket=self.tokens_bucket + other.tokens_bucket,
-        )
-
-    def turbo_mode_on(self):
-        """Set the refill rate to infinity for both buckets."""
-        self.requests_bucket.turbo_mode_on()
-        self.tokens_bucket.turbo_mode_on()
-
-    def turbo_mode_off(self):
-        """Restore the refill rate to its original value for both buckets."""
-        self.requests_bucket.turbo_mode_off()
-        self.tokens_bucket.turbo_mode_off()
-
-    @classmethod
-    def infinity_bucket(cls, model_name: str = "not_specified") -> "ModelBuckets":
-        """Create a bucket with infinite capacity and refill rate."""
-        from edsl.jobs.buckets.TokenBucket import TokenBucket
-
-        return cls(
-            requests_bucket=TokenBucket(
-                bucket_name=model_name,
-                bucket_type="requests",
-                capacity=float("inf"),
-                refill_rate=float("inf"),
-            ),
-            tokens_bucket=TokenBucket(
-                bucket_name=model_name,
-                bucket_type="tokens",
-                capacity=float("inf"),
-                refill_rate=float("inf"),
-            ),
-        )
-
-    def visualize(self):
-        """Visualize the token and request buckets."""
-        plot1 = self.requests_bucket.visualize()
-        plot2 = self.tokens_bucket.visualize()
-        return plot1, plot2
-
-    def __repr__(self):
-        return f"ModelBuckets(requests_bucket={self.requests_bucket}, tokens_bucket={self.tokens_bucket})"