mirror of
https://github.com/semantica-agi/semantica.git
synced 2026-08-29 04:26:20 +00:00
KaifAhmad1andClaude Sonnet 4.6
9a6c07417e
semantica.semantic_extract.models does not exist; Entity is defined in ner_extractor.py and exported from semantica.semantic_extract directly. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
|
|
import os
|
|
import sys
|
|
import pytest
|
|
|
|
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
|
|
from semantica.semantic_extract.methods import (
|
|
extract_entities_llm,
|
|
extract_relations_llm,
|
|
extract_triplets_llm,
|
|
)
|
|
from semantica.semantic_extract.providers import create_provider
|
|
from semantica.semantic_extract import Entity
|
|
|
|
NOVITA_API_KEY = os.environ.get("NOVITA_API_KEY")
|
|
NOVITA_MODEL = "deepseek/deepseek-v3.2"
|
|
TEXT = (
|
|
"Apple Inc. was founded by Steve Jobs, Steve Wozniak, and Ronald Wayne in 1976. "
|
|
"It is headquartered in Cupertino, California. The company designs, manufactures, "
|
|
"and markets smartphones, personal computers, tablets, wearables, and accessories."
|
|
)
|
|
|
|
pytestmark = pytest.mark.skipif(
|
|
not NOVITA_API_KEY,
|
|
reason="NOVITA_API_KEY not set",
|
|
)
|
|
|
|
|
|
def test_novita_provider_available():
|
|
provider = create_provider("novita")
|
|
assert provider.is_available(), "Novita provider not available — check NOVITA_API_KEY and openai install"
|
|
|
|
|
|
def test_novita_entity_extraction():
|
|
entities = extract_entities_llm(TEXT, provider="novita", model=NOVITA_MODEL)
|
|
assert isinstance(entities, list), "Expected a list of entities"
|
|
assert len(entities) > 0, "No entities extracted"
|
|
|
|
|
|
def test_novita_relation_extraction():
|
|
sample_entities = [
|
|
Entity(name="Apple Inc.", type="ORGANIZATION"),
|
|
Entity(name="Steve Jobs", type="PERSON"),
|
|
]
|
|
relations = extract_relations_llm(TEXT, entities=sample_entities, provider="novita", model=NOVITA_MODEL)
|
|
assert isinstance(relations, list), "Expected a list of relations"
|
|
|
|
|
|
def test_novita_triplet_extraction():
|
|
triplets = extract_triplets_llm(TEXT, provider="novita", model=NOVITA_MODEL)
|
|
assert isinstance(triplets, list), "Expected a list of triplets"
|
|
assert len(triplets) > 0, "No triplets extracted"
|
|
|
|
|
|
def test_novita_chunked_extraction():
|
|
long_text = " ".join([TEXT] * 10)
|
|
entities = extract_entities_llm(
|
|
long_text,
|
|
provider="novita",
|
|
model=NOVITA_MODEL,
|
|
max_text_length=200,
|
|
)
|
|
assert isinstance(entities, list), "Expected a list of entities from chunked extraction"
|
|
assert len(entities) > 0, "No entities extracted from chunked text"
|