mirror of https://github.com/doccano/doccano.git
pythondatasetsactive-learningtext-annotationdatasetnatural-language-processingdata-labelingmachine-learningannotation-tool
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
2.5 KiB
56 lines
2.5 KiB
import { AnnotationApplicationService } from '../annotationApplicationService'
|
|
import { RelationDTO } from './relationData'
|
|
import { SpanDTO } from './sequenceLabelingData'
|
|
import { APISequenceLabelingRepository } from '~/repositories/tasks/sequenceLabeling/apiSequenceLabeling'
|
|
import { Span } from '~/domain/models/tasks/sequenceLabeling'
|
|
import { RelationRepository } from "~/domain/models/tasks/relationRepository"
|
|
import { RelationItem } from "~/domain/models/tasks/relation"
|
|
|
|
export class SequenceLabelingApplicationService extends AnnotationApplicationService<Span> {
|
|
constructor(
|
|
readonly repository: APISequenceLabelingRepository,
|
|
readonly relationRepository: RelationRepository
|
|
) {
|
|
super(new APISequenceLabelingRepository())
|
|
}
|
|
|
|
public async list(projectId: string, docId: number): Promise<SpanDTO[]> {
|
|
const items = await this.repository.list(projectId, docId)
|
|
return items.map(item => new SpanDTO(item))
|
|
}
|
|
|
|
public async create(projectId: string, docId: number, labelId: number, startOffset: number, endOffset: number): Promise<void> {
|
|
const item = new Span(0, labelId, 0, startOffset, endOffset)
|
|
try {
|
|
await this.repository.create(projectId, docId, item)
|
|
} catch(e: any) {
|
|
console.log(e.response.data.detail)
|
|
}
|
|
}
|
|
|
|
public async changeLabel(projectId: string, docId: number, annotationId: number, labelId: number): Promise<void> {
|
|
try {
|
|
await this.repository.update(projectId, docId, annotationId, labelId)
|
|
} catch(e: any) {
|
|
console.log(e.response.data.detail)
|
|
}
|
|
}
|
|
|
|
public async listRelations(projectId: string, docId: number): Promise<RelationDTO[]> {
|
|
const items = await this.relationRepository.list(projectId, docId)
|
|
return items.map(item => new RelationDTO(item))
|
|
}
|
|
|
|
public async createRelation(projectId: string, docId: number, fromId: number, toId: number, typeId: number): Promise<void> {
|
|
const relation = new RelationItem(0, fromId, toId, typeId)
|
|
await this.relationRepository.create(projectId, docId, relation)
|
|
}
|
|
|
|
public async deleteRelation(projectId: string, docId: number, relationId: number): Promise<void> {
|
|
await this.relationRepository.delete(projectId, docId, relationId)
|
|
}
|
|
|
|
public async updateRelation(projectId: string, docId: number, relationId: number, typeId: number): Promise<void> {
|
|
await this.relationRepository.update(projectId, docId, relationId, typeId)
|
|
}
|
|
}
|