Skip to content

Commit c04a472

Browse files
committed
add line length with 80
1 parent cfcad7c commit c04a472

File tree

1 file changed

+24
-11
lines changed

1 file changed

+24
-11
lines changed

veadk/agent.py

Lines changed: 24 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,9 @@ class Agent(LlmAgent):
120120

121121
context_cache_config: Optional[ContextCacheConfig] = None
122122

123-
run_processor: Optional[BaseRunProcessor] = Field(default=None, exclude=True)
123+
run_processor: Optional[BaseRunProcessor] = Field(
124+
default=None, exclude=True
125+
)
124126
"""Optional run processor for intercepting and processing agent execution flows.
125127
126128
The run processor can be used to implement cross-cutting concerns such as:
@@ -146,7 +148,9 @@ class Agent(LlmAgent):
146148

147149
skills: list[str] = Field(default_factory=list)
148150

149-
skills_mode: Literal["skills_sandbox", "aio_sandbox", "local"] = "skills_sandbox"
151+
skills_mode: Literal["skills_sandbox", "aio_sandbox", "local"] = (
152+
"skills_sandbox"
153+
)
150154

151155
example_store: Optional[BaseExampleProvider] = None
152156

@@ -191,7 +195,8 @@ def model_post_init(self, __context: Any) -> None:
191195
if self.model_name:
192196
model_name = self.model_name[0]
193197
fallbacks = [
194-
f"{self.model_provider}/{m}" for m in self.model_name[1:]
198+
f"{self.model_provider}/{m}"
199+
for m in self.model_name[1:]
195200
]
196201
logger.info(
197202
f"Using primary model: {model_name}, with fallbacks: {self.model_name[1:]}"
@@ -244,7 +249,9 @@ def model_post_init(self, __context: Any) -> None:
244249
if hasattr(load_memory, "custom_metadata"):
245250
if not load_memory.custom_metadata:
246251
load_memory.custom_metadata = {}
247-
load_memory.custom_metadata["backend"] = self.long_term_memory.backend
252+
load_memory.custom_metadata["backend"] = (
253+
self.long_term_memory.backend
254+
)
248255
self.tools.append(load_memory)
249256

250257
if self.enable_authz:
@@ -336,9 +343,7 @@ def load_skills(self):
336343
self.instruction += "\nYou have the following skills:\n"
337344

338345
for skill in skills.values():
339-
self.instruction += (
340-
f"- name: {skill.name}\n- description: {skill.description}\n\n"
341-
)
346+
self.instruction += f"- name: {skill.name}\n- description: {skill.description}\n\n"
342347

343348
if self.skills_mode not in [
344349
"skills_sandbox",
@@ -354,11 +359,17 @@ def load_skills(self):
354359
logger.warning("No skills loaded.")
355360

356361
def _prepare_tracers(self):
357-
enable_apmplus_tracer = os.getenv("ENABLE_APMPLUS", "false").lower() == "true"
358-
enable_cozeloop_tracer = os.getenv("ENABLE_COZELOOP", "false").lower() == "true"
362+
enable_apmplus_tracer = (
363+
os.getenv("ENABLE_APMPLUS", "false").lower() == "true"
364+
)
365+
enable_cozeloop_tracer = (
366+
os.getenv("ENABLE_COZELOOP", "false").lower() == "true"
367+
)
359368
enable_tls_tracer = os.getenv("ENABLE_TLS", "false").lower() == "true"
360369

361-
if not (enable_apmplus_tracer or enable_cozeloop_tracer or enable_tls_tracer):
370+
if not (
371+
enable_apmplus_tracer or enable_cozeloop_tracer or enable_tls_tracer
372+
):
362373
logger.info("No exporter enabled by env, skip prepare tracers.")
363374
return
364375

@@ -391,7 +402,9 @@ def _prepare_tracers(self):
391402
self.tracers[0].exporters.append(CozeloopExporter()) # type: ignore
392403
logger.info("Enable CozeLoop exporter by env.")
393404

394-
if enable_tls_tracer and not any(isinstance(e, TLSExporter) for e in exporters):
405+
if enable_tls_tracer and not any(
406+
isinstance(e, TLSExporter) for e in exporters
407+
):
395408
self.tracers[0].exporters.append(TLSExporter()) # type: ignore
396409
logger.info("Enable TLS exporter by env.")
397410

0 commit comments

Comments
 (0)