41 lines
1.0 KiB
Python
41 lines
1.0 KiB
Python
import asyncio
|
|
from services.llm_providers.gemini_provider import gemini_structured_json_response
|
|
|
|
async def test_simple_schema():
|
|
"""Test with a very simple schema to see if structured output works at all"""
|
|
|
|
# Very simple schema
|
|
simple_schema = {
|
|
"type": "object",
|
|
"properties": {
|
|
"name": {"type": "string"},
|
|
"age": {"type": "integer"}
|
|
}
|
|
}
|
|
|
|
simple_prompt = "Generate a person with a name and age."
|
|
|
|
print("Testing simple schema...")
|
|
print(f"Schema: {simple_schema}")
|
|
print(f"Prompt: {simple_prompt}")
|
|
print("\n" + "="*50 + "\n")
|
|
|
|
try:
|
|
result = gemini_structured_json_response(
|
|
prompt=simple_prompt,
|
|
schema=simple_schema,
|
|
temperature=0.3,
|
|
max_tokens=100
|
|
)
|
|
|
|
print("Result:")
|
|
print(result)
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
import traceback
|
|
traceback.print_exc()
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(test_simple_schema())
|