Skip to content

Commit c98af4b

Browse files
committed
feat: use lf_toolkit schema
1 parent e13f95d commit c98af4b

File tree

3 files changed

+3
-120
lines changed

3 files changed

+3
-120
lines changed

requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -10,4 +10,5 @@ langdetect
1010
langgraph
1111
langsmith
1212

13+
lf_toolkit[ipc] @ git+https://github.com/lambda-feedback/toolkit-python.git@main
1314
pytest

src/module.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,13 @@
1+
import time
12
from typing import Any
3+
from lf_toolkit.chat import ChatResult as Result, ChatParams as Params
24

35
try:
4-
from .module_response import Result, Params
56
from .agents.utils.parse_json_to_prompt import parse_json_to_prompt
67
from .agents.base_agent.base_agent import invoke_base_agent
78
except ImportError:
8-
from src.module_response import Result, Params
99
from src.agents.utils.parse_json_to_prompt import parse_json_to_prompt
1010
from src.agents.base_agent.base_agent import invoke_base_agent
11-
import time
1211

1312
def chat_module(message: Any, params: Params) -> Result:
1413
"""

src/module_response.py

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

0 commit comments

Comments
 (0)