python-agents-examples/basics/function_calling.py at main · archytus10/python-agents-examples · GitHub

History

49 lines (39 loc) · 1.52 KB

Raw

## This is a basic example of how to use function calling.

## To test the function, you can ask the agent to print to the console!

import logging

from pathlib import Path

from dotenv import load_dotenv

from livekit.agents import JobContext, WorkerOptions, cli

from livekit.agents.llm import function_tool

from livekit.agents.voice import Agent, AgentSession, RunContext

from livekit.plugins import deepgram, openai, silero

logger = logging.getLogger("function-calling")

logger.setLevel(logging.INFO)

load_dotenv(dotenv_path=Path(__file__).parent.parent / '.env')

class FunctionAgent(Agent):

def __init__(self) -> None:

super().__init__(

instructions="""

You are a helpful assistant communicating through voice. Don't use any unpronouncable characters.

Note: If asked to print to the console, use the `print_to_console` function.

""",

stt=deepgram.STT(),

llm=openai.LLM(model="gpt-4o"),

tts=openai.TTS(),

vad=silero.VAD.load()

)

@function_tool

async def print_to_console(self, context: RunContext):

print("Console Print Success!")

return None, "I've printed to the console."

async def on_enter(self):

self.session.generate_reply()

async def entrypoint(ctx: JobContext):

await ctx.connect()

session = AgentSession()

await session.start(

agent=FunctionAgent(),

room=ctx.room

)

if __name__ == "__main__":

cli.run_app(WorkerOptions(entrypoint_fnc=entrypoint))

python-agents-examples/basics/function_calling.py at main · archytus10/python-agents-examples · GitHub

Expand file tree

Breadcrumbs

function_calling.py

Latest commit

History

Breadcrumbs

function_calling.py

File metadata and controls