Tangledup-ai/lang-agent
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make thread_id a string

Browse Source
This commit is contained in:
goulustis
2025-12-30 16:11:44 +08:00
parent 42a6f057e4
commit a8c2dd7723
4 changed files with 5 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
fastapi_server/fake_stream_server_dashscopy.py
View File

@@ -33,7 +33,7 @@ class DSApplicationCallRequest(BaseModel):
messages: List[DSMessage] messages: List[DSMessage]
stream: bool = Field(default=True) stream: bool = Field(default=True)
# Optional overrides for pipeline behavior # Optional overrides for pipeline behavior
thread_id: Optional[int] = Field(default=3) thread_id: Optional[str] = Field(default="3")
app = FastAPI(title="DashScope-Compatible Application API", app = FastAPI(title="DashScope-Compatible Application API",

2
fastapi_server/server_dashscope.py
View File

@@ -33,7 +33,7 @@ class DSApplicationCallRequest(BaseModel):
messages: List[DSMessage] messages: List[DSMessage]
stream: bool = Field(default=True) stream: bool = Field(default=True)
# Optional overrides for pipeline behavior # Optional overrides for pipeline behavior
thread_id: Optional[int] = Field(default=3) thread_id: Optional[str] = Field(default="3")
app = FastAPI(title="DashScope-Compatible Application API", app = FastAPI(title="DashScope-Compatible Application API",

2
fastapi_server/server_openai.py
View File

@@ -33,7 +33,7 @@ class OpenAIChatCompletionRequest(BaseModel):
temperature: Optional[float] = Field(default=1.0) temperature: Optional[float] = Field(default=1.0)
max_tokens: Optional[int] = Field(default=None) max_tokens: Optional[int] = Field(default=None)
# Optional overrides for pipeline behavior # Optional overrides for pipeline behavior
thread_id: Optional[int] = Field(default=3) thread_id: Optional[str] = Field(default='3')
app = FastAPI( app = FastAPI(

6
lang_agent/pipeline.py
View File

@@ -145,7 +145,7 @@ class Pipeline:
async for chunk in out: async for chunk in out:
yield chunk yield chunk
def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:int = None): def chat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:str = '3'):
""" """
as_stream (bool): if true, enable the thing to be streamable as_stream (bool): if true, enable the thing to be streamable
as_raw (bool): return full dialoge of List[SystemMessage, HumanMessage, ToolMessage] as_raw (bool): return full dialoge of List[SystemMessage, HumanMessage, ToolMessage]
@@ -153,7 +153,6 @@ class Pipeline:
# NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph # NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph
u = DEFAULT_PROMPT u = DEFAULT_PROMPT
thread_id = thread_id if thread_id is not None else 3
inp = {"messages":[SystemMessage(u), inp = {"messages":[SystemMessage(u),
HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id}} HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id}}
@@ -188,7 +187,7 @@ class Pipeline:
assert 0, "something is wrong" assert 0, "something is wrong"
async def achat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:int = None): async def achat(self, inp:str, as_stream:bool=False, as_raw:bool=False, thread_id:str = '3'):
""" """
Async version of chat using LangGraph's native async support. Async version of chat using LangGraph's native async support.
@@ -198,7 +197,6 @@ class Pipeline:
# NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph # NOTE: this prompt will be overwritten by 'configs/route_sys_prompts/chat_prompt.txt' for route graph
u = DEFAULT_PROMPT u = DEFAULT_PROMPT
thread_id = thread_id if thread_id is not None else 3
inp_data = {"messages":[SystemMessage(u), inp_data = {"messages":[SystemMessage(u),
HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id}} HumanMessage(inp)]}, {"configurable": {"thread_id": thread_id}}