decorator based tool-use

[aniketmaurya]

Before submitting

• Was this discussed/agreed via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure to update the docs?
• Did you write any new necessary tests?

What does this PR do?

Users will be able to perform tool-use using a simple decorator based definition.

from litai import LLM, tool

@tool
def get_weather(location: str):
    return f"The weather in {location} is sunny"

llm = LLM(model="openai/gpt-4")
response = llm.chat("What's the weather in Tokyo?", tools=[get_weather])
result = llm.call_tool(response, tools=[get_weather])
print(result)

PR review

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in GitHub issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

[aniketmaurya]

looking into the failing CI

[williamFalcon]

let's use this syntax so we don't need 2 lines of imports...

from litai import LLM, tool

@tool
def get_weather(location: str):
    return f"The weather in {location} is sunny"
...