Skip to content

Commit

Permalink
Deleted stale code
Browse files Browse the repository at this point in the history
  • Loading branch information
albertkimjunior committed Jun 26, 2024
1 parent 9e5b0d5 commit a5a33df
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 18 deletions.
15 changes: 0 additions & 15 deletions agentops/enums.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,21 +8,6 @@ class EventType(Enum):
TOOL = "tools"
ERROR = "errors"


class Models(Enum):
GPT_3_5_TURBO = "gpt-3.5-turbo"
GPT_3_5_TURBO_0301 = "gpt-3.5-turbo-0301"
GPT_3_5_TURBO_0613 = "gpt-3.5-turbo-0613"
GPT_3_5_TURBO_16K = "gpt-3.5-turbo-16k"
GPT_3_5_TURBO_16K_0613 = "gpt-3.5-turbo-16k-0613"
GPT_4_0314 = "gpt-4-0314"
GPT_4 = "gpt-4"
GPT_4_32K = "gpt-4-32k"
GPT_4_32K_0314 = "gpt-4-32k-0314"
GPT_4_0613 = "gpt-4-0613"
TEXT_EMBEDDING_ADA_002 = "text-embedding-ada-002"


class EndState(Enum):
SUCCESS = "Success"
FAIL = "Fail"
Expand Down
6 changes: 3 additions & 3 deletions agentops/event.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,10 @@
Event: Represents discrete events to be recorded.
"""

from dataclasses import asdict, dataclass, field
from dataclasses import dataclass, field
from typing import Any, Dict, List, Optional, Sequence, Union
from .helpers import get_ISO_time, check_call_stack_for_agent_id
from .enums import EventType, Models
from .enums import EventType
from uuid import UUID, uuid4
import traceback

Expand Down Expand Up @@ -72,7 +72,7 @@ class LLMEvent(Event):
prompt_tokens(int, optional): The number of tokens in the prompt message.
completion(str, object, optional): The message or returned by the LLM. Preferably in ChatML format which is more fully supported by AgentOps.
completion_tokens(int, optional): The number of tokens in the completion message.
model(Models, str, optional): LLM model e.g. "gpt-4". Models defined in enums.Models are more fully supported by AgentOps e.g. extra features in dashboard.
model(str, optional): LLM model e.g. "gpt-4".
"""

Expand Down

0 comments on commit a5a33df

Please sign in to comment.