pip install seekrai langchain langchain-community

# Standard library
import getpass
import os

# Third-party
from langchain.promptsimport ChatPromptTemplate
from langchain.schemaimport HumanMessage
from langchain_core.runnablesimport RunnableSequence

# OSS SeekrFlow integration
from langchain_seekrflowimport ChatSeekrFlow
from seekraiimport SeekrFlow

SEEKR_API_KEY="your-api-key-here"

os.environ["SEEKR_API_KEY"]= getpass.getpass("Enter your Seekr API key:")

os.environ["SEEKR_API_KEY"]
seekr_client= SeekrFlow(api_key=SEEKR_API_KEY)

llm= ChatSeekrFlow(
    client=seekr_client, model_name="meta-llama/Meta-Llama-3-8B-Instruct"
)

response= llm.invoke([HumanMessage(content="Hello, Seekr!")])
print(response.content)

Hello there! I'm Seekr, nice to meet you! What brings you here today? Do you have a question, or are you looking for some help with something? I'm all ears (or rather, all text)!

prompt= ChatPromptTemplate.from_template("Translate to French: {text}")

chain: RunnableSequence= prompt| llm
result= chain.invoke({"text":"Good morning"})
print(result)

content='The translation of "Good morning" in French is:\n\n"Bonne journée"' additional_kwargs={} response_metadata={}

deftest_stream():
"""Test synchronous invocation in streaming mode."""
print("\n🔹 Testing Sync `stream()` (Streaming)...")

for chunkin llm.stream([HumanMessage(content="Write me a haiku.")]):
print(chunk.content, end="", flush=True)


# ✅ Ensure streaming is enabled
llm= ChatSeekrFlow(
    client=seekr_client,
    model_name="meta-llama/Meta-Llama-3-8B-Instruct",
    streaming=True,# ✅ Enable streaming
)

# ✅ Run sync streaming test
test_stream()

🔹 Testing Sync `stream()` (Streaming)...
Here is a haiku:

Golden sunset fades
Ripples on the quiet lake
Peaceful evening sky

# Define a minimal mock SeekrFlow client
classMockSeekrClient:
"""Mock SeekrFlow API client that mimics the real API structure."""

classMockChat:
"""Mock Chat object with a completions method."""

classMockCompletions:
"""Mock Completions object with a create method."""

defcreate(self,*args,**kwargs):
return{
"choices":[{"message":{"content":"Mock response"}}]
}# Mimic API response

        completions= MockCompletions()

    chat= MockChat()


deftest_initialization_errors():
"""Test that invalid ChatSeekrFlow initializations raise expected errors."""

    test_cases=[
{
"name":"Missing Client",
"args":{"client":None,"model_name":"seekrflow-model"},
"expected_error":"SeekrFlow client cannot be None.",
},
{
"name":"Missing Model Name",
"args":{"client": MockSeekrClient(),"model_name":""},
"expected_error":"A valid model name must be provided.",
},
]

for testin test_cases:
try:
print(f"Running test:{test['name']}")
            faulty_llm= ChatSeekrFlow(**test["args"])

# If no error is raised, fail the test
print(f"❌ Test '{test['name']}' failed: No error was raised!")
except Exceptionas e:
            error_msg=str(e)
assert test["expected_error"]in error_msg,f"Unexpected error:{error_msg}"
print(f"✅ Expected Error:{error_msg}")


# Run test
test_initialization_errors()

Running test: Missing Client
✅ Expected Error: SeekrFlow client cannot be None.
Running test: Missing Model Name
✅ Expected Error: A valid model name must be provided.