modified test script

tests/test_app.py

@@ -1,10 +1,28 @@
-import sys, os
-sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
+import types
+import app
 
-from app import chat
 
-def test_chat_response():
-    user_input = "I'm buying a plastic bottle of water."
-    response = chat(user_input)
-    assert isinstance(response, str)
-    assert len(response) > 0
+def test_respond_function_exists():
+    """Check that the app has a respond() function."""
+    assert hasattr(app, "respond")
+    assert callable(app.respond)
+
+
+def test_respond_returns_generator():
+    """respond() should return a generator when called with minimal args."""
+    # Fake OAuthToken object for testing (since we don't want to call real HF API in CI)
+    class DummyToken:
+        token = "dummy"
+
+    gen = app.respond(
+        message="I'm buying a bottle of water.",
+        history=[],
+        system_message="You are Sustainable.ai.",
+        max_tokens=10,
+        temperature=0.7,
+        top_p=0.9,
+        hf_token=DummyToken(),
+    )
+
+    # respond() is a generator, not a plain string
+    assert isinstance(gen, types.GeneratorType)