SambaNova

import os
import json
from datetime import datetime
from linkup import LinkupClient
from sambanova import SambaNova

linkup_client = LinkupClient(api_key=os.environ.get("LINKUP_API_KEY"))
sambanova_client = SambaNova(api_key=os.environ.get("SAMBANOVA_API_KEY"))

tools = [{
    "type": "function",
    "function": {
        "name": "search_web",
        "description": "Search the web for current or verifiable information. Make the query specific: say what to find and which facts to return. Return source URLs. If no reliable source is found, say so.",
        "parameters": {
            "type": "object",
            "properties": {
                "query": {
                    "type": "string",
                    "description": "The search query"
                }
            },
            "required": ["query"]
        }
    }
}]

def main():
    print("--- SambaNova + Linkup ---")
    print("Type 'quit' to exit.\n")

    today_str = datetime.now().strftime("%B %d, %Y")
    system_prompt = (
        f"You are a helpful assistant. Today is {today_str}. "
        f"Use web search for current or verifiable information. "
        f"Prefer searching when facts could be outdated, and cite source URLs."
    )

    history = [{"role": "system", "content": system_prompt}]

    while True:
        try:
            user_input = input("You: ")
            if user_input.lower() in ["quit", "exit"]:
                print("Goodbye!")
                break

            history.append({"role": "user", "content": user_input})

            response = sambanova_client.chat.completions.create(
                model="gpt-oss-120b",
                messages=history,
                tools=tools
            )
            message = response.choices[0].message

            while message.tool_calls:
                history.append(message)
                for tc in message.tool_calls:
                    q = json.loads(tc.function.arguments)["query"]
                    print(f"Searching using Linkup: {q}...")
                    try:
                        result = linkup_client.search(
                            query=q,
                            depth="standard",
                            output_type="sourcedAnswer"
                        )
                        content = json.dumps(result.model_dump())
                    except Exception as e:
                        content = f"Search error: {e}"
                    history.append({
                        "role": "tool",
                        "name": "search_web",
                        "tool_call_id": tc.id,
                        "content": content
                    })

                response = sambanova_client.chat.completions.create(
                    model="gpt-oss-120b",
                    messages=history,
                    tools=tools
                )
                message = response.choices[0].message

            print(f"Agent: {message.content}\n")
            history.append(message)

        except Exception as e:
            print(f"Error: {e}")

if __name__ == "__main__":
    main()