Skip to content

Commit

Permalink
add edit test case endpoint
Browse files Browse the repository at this point in the history
  • Loading branch information
jalexanderII committed Jul 2, 2024
1 parent 5dcb137 commit 0027de6
Show file tree
Hide file tree
Showing 6 changed files with 11 additions and 10 deletions.
8 changes: 4 additions & 4 deletions parea/client.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@
from typing import Any, AsyncIterable, Callable, Dict, Iterable, List, Optional, Union

import asyncio
import logging
import os
import time
from typing import Any, AsyncIterable, Callable, Dict, Iterable, List, Optional, Union

import httpx
from attrs import asdict, define, field
Expand All @@ -13,8 +14,7 @@
from parea.cache.cache import Cache
from parea.constants import PAREA_OS_ENV_EXPERIMENT_UUID
from parea.experiment.datasets import create_test_cases, create_test_collection
from parea.helpers import gen_trace_id, serialize_metadata_values, structure_trace_log_from_api, \
structure_trace_logs_from_api
from parea.helpers import gen_trace_id, serialize_metadata_values, structure_trace_log_from_api, structure_trace_logs_from_api
from parea.parea_logger import parea_logger
from parea.schemas import EvaluationResult
from parea.schemas.models import (
Expand All @@ -34,9 +34,9 @@
TestCaseCollection,
TraceLogFilters,
TraceLogTree,
UpdateTestCase,
UseDeployedPrompt,
UseDeployedPromptResponse,
UpdateTestCase,
)
from parea.utils.trace_utils import get_current_trace_id, get_root_trace_id, logger_all_possible, trace_data

Expand Down
2 changes: 1 addition & 1 deletion parea/cookbook/async_enpoints_for_datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
from dotenv import load_dotenv

from parea import Parea
from parea.schemas import TestCaseCollection, TestCase, UpdateTestCase
from parea.schemas import TestCase, TestCaseCollection, UpdateTestCase

load_dotenv()

Expand Down
2 changes: 1 addition & 1 deletion parea/cookbook/enpoints_for_datasets.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from dotenv import load_dotenv

from parea import Parea
from parea.schemas import TestCaseCollection, TestCase, UpdateTestCase
from parea.schemas import TestCase, TestCaseCollection, UpdateTestCase

load_dotenv()

Expand Down
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from typing import List, Tuple

import os
from datetime import datetime
from typing import List, Tuple

from dotenv import load_dotenv
from openai import OpenAI
Expand Down
3 changes: 2 additions & 1 deletion parea/schemas/models.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from typing import Any, Dict, Iterable, List, Optional, Tuple, Union

import json
from enum import Enum
from typing import Any, Dict, Iterable, List, Optional, Tuple, Union

from attrs import define, field, validators

Expand Down
3 changes: 1 addition & 2 deletions parea/wrapper/anthropic/anthropic.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,7 @@

from parea.cache.cache import Cache
from parea.helpers import timezone_aware_now
from parea.schemas import CacheRequest, LLMInputs
from parea.schemas import ModelParams
from parea.schemas import CacheRequest, LLMInputs, ModelParams
from parea.schemas import Role as PareaRole
from parea.schemas import TraceLog
from parea.utils.trace_utils import make_output, trace_data
Expand Down

0 comments on commit 0027de6

Please sign in to comment.