jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Fix tests

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-28 13:17:40 -03:00
commit d0f10e107e
3 changed files with 6 additions and 119 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/api/v1/callback.py
View file

@ -14,6 +14,11 @@ from loguru import logger
class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler): class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler):
"""Callback handler for streaming LLM responses.""" """Callback handler for streaming LLM responses."""
@property
def ignore_chain(self) -> bool:
"""Whether to ignore chain callbacks."""
return False
def __init__(self, client_id: str): def __init__(self, client_id: str):
self.chat_service = get_chat_service() self.chat_service = get_chat_service()
self.client_id = client_id self.client_id = client_id

18
src/backend/langflow/services/chat/utils.py
View file

@ -3,12 +3,10 @@ from typing import Any
from langchain.agents import AgentExecutor from langchain.agents import AgentExecutor
from langchain.chains.base import Chain from langchain.chains.base import Chain
from langchain_core.runnables import Runnable from langchain_core.runnables import Runnable
from loguru import logger
from langflow.api.v1.schemas import ChatMessage from langflow.api.v1.schemas import ChatMessage
from langflow.interface.utils import try_setting_streaming_options from langflow.interface.utils import try_setting_streaming_options
from langflow.processing.base import get_result_and_steps from langflow.processing.base import get_result_and_steps
from langflow.utils.chat import ChatDefinition from loguru import logger
LANGCHAIN_RUNNABLES = (Chain, Runnable, AgentExecutor) LANGCHAIN_RUNNABLES = (Chain, Runnable, AgentExecutor)
@ -40,20 +38,6 @@ async def process_graph(
client_id=client_id, client_id=client_id,
session_id=session_id, session_id=session_id,
) )
elif isinstance(build_result, ChatDefinition):
raw_output = await run_build_result(
build_result,
chat_inputs,
client_id=client_id,
session_id=session_id,
)
if isinstance(raw_output, dict):
if not build_result.output_key:
raise ValueError("No output key provided to ChatDefinition when returning a dict.")
result = raw_output[build_result.output_key]
else:
result = raw_output
intermediate_steps = []
else: else:
raise TypeError(f"Unknown type {type(build_result)}") raise TypeError(f"Unknown type {type(build_result)}")
logger.debug("Generated result and intermediate_steps") logger.debug("Generated result and intermediate_steps")

102
tests/test_prompts_template.py
View file

@ -1,5 +1,4 @@
from fastapi.testclient import TestClient from fastapi.testclient import TestClient
from langflow.services.deps import get_settings_service from langflow.services.deps import get_settings_service
@ -10,104 +9,3 @@ def test_prompts_settings(client: TestClient, logged_in_headers):
json_response = response.json() json_response = response.json()
prompts = json_response["prompts"] prompts = json_response["prompts"]
assert set(prompts.keys()) == set(settings_service.settings.PROMPTS) assert set(prompts.keys()) == set(settings_service.settings.PROMPTS)
def test_prompt_template(client: TestClient, logged_in_headers):
response = client.get("api/v1/all", headers=logged_in_headers)
assert response.status_code == 200
json_response = response.json()
prompts = json_response["prompts"]
prompt = prompts["PromptTemplate"]
template = prompt["template"]
assert template["input_variables"] == {
"required": True,
"dynamic": True,
"placeholder": "",
"show": False,
"multiline": False,
"password": False,
"name": "input_variables",
"type": "str",
"list": True,
"advanced": False,
"info": "",
"fileTypes": [],
}
assert template["output_parser"] == {
"required": False,
"dynamic": True,
"placeholder": "",
"show": False,
"multiline": False,
"password": False,
"name": "output_parser",
"type": "BaseOutputParser",
"list": False,
"advanced": False,
"info": "",
"fileTypes": [],
}
assert template["partial_variables"] == {
"required": False,
"dynamic": True,
"placeholder": "",
"show": False,
"multiline": False,
"password": False,
"name": "partial_variables",
"type": "dict",
"list": False,
"advanced": False,
"info": "",
"fileTypes": [],
}
assert template["template"] == {
"required": True,
"dynamic": True,
"placeholder": "",
"show": True,
"multiline": True,
"password": False,
"name": "template",
"type": "prompt",
"list": False,
"advanced": False,
"info": "",
"fileTypes": [],
}
assert template["template_format"] == {
"required": False,
"dynamic": True,
"placeholder": "",
"show": False,
"multiline": False,
"value": "f-string",
"password": False,
"name": "template_format",
"type": "str",
"list": False,
"advanced": False,
"info": "",
"fileTypes": [],
}
assert template["validate_template"] == {
"required": False,
"dynamic": True,
"placeholder": "",
"show": False,
"multiline": False,
"value": False,
"password": False,
"name": "validate_template",
"type": "bool",
"list": False,
"advanced": False,
"info": "",
"fileTypes": [],
}