Skip to content

Commit de306fa

Browse files
taoyifan89taoyifan89
authored
Change package name.
Change-Id: I1c9227ed9f397e66d1f4e449ef87a51496ddc8ce
1 parent 7dcd225 commit de306fa

22 files changed

Lines changed: 36 additions & 60 deletions

File tree

‎README.md‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
# Loop Python SDK
1+
# CozeLoop Python SDK
22
English | [简体中文](README.zh_CN.md)
33

44
## Overview
55

6-
The Loop SDK is a Python client for interacting with [CozeLoop platform](https://loop.coze.cn).
6+
The CozeLoop SDK is a Python client for interacting with [CozeLoop platform](https://loop.coze.cn).
77
Key features:
88
- Report trace
99
- Get and format prompt

‎README.zh_CN.md‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
1-
# Loop Python SDK
1+
# CozeLoop Python SDK
22
[English](README.md) | 简体中文
33

44
## 概述
55

6-
Loop SDK 是一个用于与 [CozeLoop 平台](https://loop.coze.cn) 进行交互的 Python 客户端。
6+
CozeLoop SDK 是一个用于与 [CozeLoop 平台](https://loop.coze.cn) 进行交互的 Python 客户端。
77
主要功能：
88
- Trace上报
99
- Prompt拉取

‎cozeloop/__init__.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
# SPDX-License-Identifier: MIT
33

44
from .entities.prompt import Prompt, Message
5-
import cozeloop.entities
65
from .logger import set_log_level, add_log_handler
76

87
from .internal import __version__

‎cozeloop/attribute/trace/__init__.py‎

Lines changed: 0 additions & 3 deletions
This file was deleted.

‎cozeloop/internal/consts/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
from contextvars import ContextVar
55

66
from .error import *
7-
from ...attribute.trace.span_key import INPUT_TOKENS, OUTPUT_TOKENS, TOKENS, INPUT, OUTPUT
7+
from ...spec.tracespce.span_key import INPUT_TOKENS, OUTPUT_TOKENS, TOKENS, INPUT, OUTPUT
88

99
# default values for loop client
1010
# ComBaseURL = "https://api.coze.com"

‎cozeloop/internal/prompt/converter.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
from typing import List, Dict
55

6-
from cozeloop.attribute.trace.prompt import PromptInput, PromptOutput, ModelMessage, PromptArgument
6+
from cozeloop.spec.tracespce import PromptInput, PromptOutput, ModelMessage, PromptArgument
77
from cozeloop.entities.prompt import (
88
Prompt as EntityPrompt,
99
Message as EntityMessage,

‎cozeloop/internal/prompt/prompt.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,7 @@
77
from jinja2 import Environment, BaseLoader, Undefined
88
from jinja2.utils import missing, object_type_repr
99

10-
from cozeloop.attribute.trace.span_key import PROMPT_KEY, INPUT, PROMPT_VERSION
11-
from cozeloop.attribute.trace.span_value import V_SCENE_PROMPT_TEMPLATE, V_SCENE_PROMPT_HUB
10+
from cozeloop.spec.tracespce import PROMPT_KEY, INPUT, PROMPT_VERSION, V_SCENE_PROMPT_TEMPLATE, V_SCENE_PROMPT_HUB
1211
from cozeloop.entities.prompt import (Prompt, Message, VariableDef, VariableType, TemplateType, Role,
1312
PromptVariable)
1413
from cozeloop.internal import consts

‎cozeloop/internal/trace/exporter.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99

1010
from typing import Dict, List, Optional, Tuple, Callable, Any
1111

12-
from cozeloop.attribute.trace.model import ModelInput, ModelMessagePart, ModelMessagePartType, ModelImageURL, ModelFileURL, ModelOutput
12+
from cozeloop.spec.tracespce import ModelInput, ModelMessagePart, ModelMessagePartType, ModelImageURL, ModelFileURL, ModelOutput
1313
from cozeloop.internal.consts import *
1414
from cozeloop.internal.httpclient import Client, BaseResponse
1515
from cozeloop.internal.trace.model.model import UploadType, Attachment, ObjectStorage

‎cozeloop/internal/trace/noop_span.py‎

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def set_input(self, input: Any) -> None:
6161
def set_output(self, output: Any) -> None:
6262
pass
6363

64-
def set_error(self, err: str) -> None:
64+
def set_error(self, err: Exception) -> None:
6565
pass
6666

6767
def set_status_code(self, code: int) -> None:
@@ -88,19 +88,13 @@ def set_thread_id_baggage(self, thread_id: str) -> None:
8888
def set_prompt(self, prompt: Prompt) -> None:
8989
pass
9090

91-
def set_prompt_baggage(self, prompt: Prompt) -> None:
92-
pass
93-
9491
def set_model_provider(self, model_provider: str) -> None:
9592
pass
9693

97-
def set_model_provider_baggage(self, model_provider: str) -> None:
98-
pass
99-
10094
def set_model_name(self, model_name: str) -> None:
10195
pass
10296

103-
def set_model_name_baggage(self, model_name: str) -> None:
97+
def set_model_call_options(self, call_options: Any) -> None:
10498
pass
10599

106100
def set_input_tokens(self, input_tokens: int) -> None:

‎cozeloop/internal/trace/span.py‎

Lines changed: 8 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -10,10 +10,10 @@
1010
import urllib.parse
1111

1212
from cozeloop import span
13-
from cozeloop.attribute.trace.model import ModelInput, ModelOutput, ModelMessagePartType, ModelMessage, ModelMessagePart, ModelImageURL, ModelFileURL, ModelChoice
14-
from cozeloop.attribute.trace.runtime import Runtime
15-
from cozeloop.attribute.trace.span_key import ERROR, PROMPT_KEY, PROMPT_VERSION, MODEL_PROVIDER, MODEL_NAME, RUNTIME_
16-
from cozeloop.attribute.trace.span_value import V_SCENE_CUSTOM, V_LANG_PYTHON
13+
from cozeloop.spec.tracespce import (ModelInput, ModelOutput, ModelMessagePartType, ModelMessage, ModelMessagePart,
14+
ModelImageURL, ModelFileURL, ModelChoice, Runtime, ERROR, PROMPT_KEY,
15+
PROMPT_VERSION, MODEL_PROVIDER, MODEL_NAME, RUNTIME_, CALL_OPTIONS,
16+
V_SCENE_CUSTOM, V_LANG_PYTHON)
1717
from cozeloop.entities.prompt import Prompt
1818
from cozeloop.internal.consts import *
1919
from cozeloop.internal.trace.noop_span import NoopSpan
@@ -293,8 +293,8 @@ def get_model_output_bytes_size(self, m_content):
293293
except Exception:
294294
return 0
295295

296-
def set_error(self, error: str):
297-
self.set_tags({ERROR: error})
296+
def set_error(self, err: Exception):
297+
self.set_tags({ERROR: err.__str__()})
298298

299299
def set_status_code(self, code: int):
300300
with self.lock:
@@ -324,23 +324,14 @@ def set_prompt(self, prompt: Prompt):
324324
if prompt.version:
325325
self.set_tags({PROMPT_VERSION: prompt.version})
326326

327-
def set_prompt_baggage(self, prompt: Prompt):
328-
if prompt.prompt_key:
329-
self.set_baggage({PROMPT_KEY: prompt.prompt_key})
330-
if prompt.version:
331-
self.set_baggage({PROMPT_VERSION: prompt.version})
332-
333327
def set_model_provider(self, model_provider: str):
334328
self.set_tags({MODEL_PROVIDER: model_provider})
335329

336-
def set_model_provider_baggage(self, model_provider: str):
337-
self.set_baggage({MODEL_PROVIDER: model_provider})
338-
339330
def set_model_name(self, model_name: str):
340331
self.set_tags({MODEL_NAME: model_name})
341332

342-
def set_model_name_baggage(self, model_name: str):
343-
self.set_baggage({MODEL_NAME: model_name})
333+
def set_model_call_options(self, model_call_options: Any):
334+
self.set_tags({CALL_OPTIONS: model_call_options})
344335

345336
def set_input_tokens(self, input_tokens: int):
346337
self.set_tags({INPUT_TOKENS: input_tokens})

0 commit comments

Comments
 (0)