就八亿参数！TeapotLLM，让AI告别“胡言乱语” 原创

!pip install teapotai

from teapotai import TeapotAI

context = """
The Eiffel Tower is a wrought iron lattice tower in Paris, France. It was designed by Gustave Eiffel and completed in 1889.
It stands at a height of 330 meters and is one of the most recognizable structures in the world.
"""

teapot_ai = TeapotAI()

answer = teapot_ai.query(
   query="What is the height of the Eiffel Tower?",
   cnotallow=context
)
print(answer)

documents = [
   "The Eiffel Tower is located in Paris, France. It was built in 1889 and stands 330 meters tall.",
   "The Great Wall of China is a historic fortification that stretches over 13,000 miles.",
   "The Amazon Rainforest is the largest tropical rainforest in the world, covering over 5.5 million square kilometers.",
   "The Grand Canyon is a natural landmark located in Arizona, USA, carved by the Colorado River.",
   "Mount Everest is the tallest mountain on Earth, located in the Himalayas along the border between Nepal and China.",
   "The Colosseum in Rome, Italy, is an ancient amphitheater known for its gladiator battles.",
   "The Sahara Desert is the largest hot desert in the world, located in North Africa.",
   "The Nile River is the longest river in the world, flowing through northeastern Africa.",
   "The Empire State Building is an iconic skyscraper in New York City that was completed in 1931 and stands at 1454 feet tall."
]

teapot_ai = TeapotAI(documents=documents)

answer = teapot_ai.chat([
   {
       "role":"system",
       "content": "You are an agent designed to answer facts about famous landmarks."
   },
   {
       "role":"user",
       "content": "What landmark was constructed in the 1800s?"
   }
])
print(answer)

from pydantic import BaseModel, Field

apartment_description = """
This spacious 2-bedroom apartment is available for rent in downtown New York. The monthly rent is $2500.
It includes 1 bathrooms and a fully equipped kitchen with modern appliances. There is also a swimming pool at the backyard and beside the building.
Pets are welcome!
Please reach out to us at 555-123-4567 or john@realty.com
"""

class ApartmentInfo(BaseModel):
   rent: float = Field(..., descriptinotallow="the monthly rent in dollars")
   bedrooms: int = Field(..., descriptinotallow="the number of bedrooms")
   bathrooms: int = Field(..., descriptinotallow="the number of bathrooms")
   phone_number: str

teapot_ai = TeapotAI()

extracted_info = teapot_ai.extract(
   ApartmentInfo,
   cnotallow=apartment_description
)
print(extracted_info)

context = """
The Great Pyramid of Giza, built around 2560 BCE, is the oldest of the Seven Wonders of the Ancient World and the only one still standing.
"""